Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dstrided_slice.h36 using strided_slice::StopForAxis; in StridedSlice()
55 const int stop_0 = StopForAxis(params_copy, input_shape, 0, start_0); in StridedSlice()
57 const int stop_1 = StopForAxis(params_copy, input_shape, 1, start_1); in StridedSlice()
59 const int stop_2 = StopForAxis(params_copy, input_shape, 2, start_2); in StridedSlice()
61 const int stop_3 = StopForAxis(params_copy, input_shape, 3, start_3); in StridedSlice()
63 const int stop_4 = StopForAxis(params_copy, input_shape, 4, start_4); in StridedSlice()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice.cc88 using ::tflite::strided_slice::StopForAxis; in CheckOutputSize()
97 int32_t end = StopForAxis(op_params, input_shape, idx, begin); in CheckOutputSize()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h122 inline int StopForAxis(const tflite::StridedSliceParams& params, in StopForAxis() function
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_strided_slice.cc64 stop_for_axis[axis] = tflite::strided_slice::StopForAxis( in StridedSlice()
Dpropagate_fixed_sizes.cc1702 int stop_index = tflite::strided_slice::StopForAxis( in ProcessStridedSliceOperator()
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc114 int32_t end = ::tflite::strided_slice::StopForAxis(op_params, input_shape, in ResizeOutputTensor()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5447 using strided_slice::StopForAxis; in StridedSlice()
5466 const int stop_0 = StopForAxis(params_copy, input_shape, 0, start_0); in StridedSlice()
5468 const int stop_1 = StopForAxis(params_copy, input_shape, 1, start_1); in StridedSlice()
5470 const int stop_2 = StopForAxis(params_copy, input_shape, 2, start_2); in StridedSlice()
5472 const int stop_3 = StopForAxis(params_copy, input_shape, 3, start_3); in StridedSlice()
5474 const int stop_4 = StopForAxis(params_copy, input_shape, 4, start_4); in StridedSlice()