Searched refs:StopForAxis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_strided_slice.cc | 64 stop_for_axis[axis] = tflite::strided_slice::StopForAxis( in StridedSlice()
|
| D | propagate_fixed_sizes.cc | 1682 int stop_index = tflite::strided_slice::StopForAxis( in ProcessStridedSliceOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 112 inline int StopForAxis(const tflite::StridedSliceParams& params, in StopForAxis() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3349 strided_slice::StopForAxis(params_copy, input_shape, 0, start_b); in StridedSlice()
3352 strided_slice::StopForAxis(params_copy, input_shape, 1, start_h); in StridedSlice()
3355 strided_slice::StopForAxis(params_copy, input_shape, 2, start_w); in StridedSlice()
3358 strided_slice::StopForAxis(params_copy, input_shape, 3, start_d); in StridedSlice()
|