Searched refs:input_shape_value (Results 1 – 2 of 2) sorted by relevance
101 Value input_shape_value = builder.create<TF::ShapeOp>( in PadForDynamicShapedInputSamePadding() local103 Value input_size_h = GetDimValue(builder, loc, input_shape_value, 1); in PadForDynamicShapedInputSamePadding()107 Value input_size_w = GetDimValue(builder, loc, input_shape_value, 2); in PadForDynamicShapedInputSamePadding()
581 auto input_shape_value = input_shape[input_shape_idx].static_size(); in ComputeShapeFromExpand() local584 input_shape_value == reshape_value || input_shape_value == 1 || in ComputeShapeFromExpand()588 std::max(input_shape_value, reshape_value)); in ComputeShapeFromExpand()798 auto input_shape_value = input_shape.value().sizes(); in ProcessShapeForConcatNode() local799 auto shape_symbol = input_shape_value.value()[idx]; in ProcessShapeForConcatNode()820 auto input_shape_value = input_shape.value().sizes(); in ProcessShapeForConcatNode() local821 auto shape_symbol = input_shape_value.value()[idx]; in ProcessShapeForConcatNode()