File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -297,7 +297,7 @@ def infer_shape_for_op_supplementary(op):
297
297
return set_shape_from_inputs_broadcast (op .inputs , op .outputs [0 ])
298
298
299
299
if op .type == "RandomUniform" :
300
- shape_op = op .inputs [0 ]
300
+ shape_op = op .inputs [0 ]. op
301
301
if not shape_op or shape_op .type != "Shape" :
302
302
return False
303
303
return set_shape_from_input (shape_op .inputs [0 ], op .outputs [0 ])
@@ -349,7 +349,7 @@ def infer_shape_for_op_supplementary(op):
349
349
if op .type == "ExpandDims" :
350
350
# https://www.tensorflow.org/api_docs/python/tf/expand_dims
351
351
input_shape = utils .get_shape_from_tf_output (op .inputs [0 ])
352
- dim_op = op .inputs [1 ]
352
+ dim_op = op .inputs [1 ]. op
353
353
if input_shape is None or not utils .is_const_op (dim_op ):
354
354
return False
355
355
You can’t perform that action at this time.
0 commit comments