Home
last modified time | relevance | path

Searched refs:input_shape_axis_value (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp1046 auto input_shape_axis_value = input_shape[axis].static_size(); in ComputeShapeForSlice() local
1050 if (cur_start < -input_shape_axis_value) { in ComputeShapeForSlice()
1053 cur_start = input_shape_axis_value + cur_start; in ComputeShapeForSlice()
1054 } else if (cur_start > input_shape_axis_value - 1) { in ComputeShapeForSlice()
1055 cur_start = input_shape_axis_value; in ComputeShapeForSlice()
1057 if (cur_end < -input_shape_axis_value) { in ComputeShapeForSlice()
1060 cur_end = input_shape_axis_value + cur_end; in ComputeShapeForSlice()
1061 } else if (cur_end > input_shape_axis_value - 1) { in ComputeShapeForSlice()
1062 cur_end = input_shape_axis_value; in ComputeShapeForSlice()