Searched refs:end_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice_test.cc | 60 const int* end_shape, const int* strides_shape, in TestStridedSliceFloat() argument
69 TfLiteIntArray* end_dims = IntArrayFromInts(end_shape); in TestStridedSliceFloat()
91 const int* input_shape, const int* begin_shape, const int* end_shape, in TestStridedSliceQuantized() argument
98 TfLiteIntArray* end_dims = IntArrayFromInts(end_shape); in TestStridedSliceQuantized()
129 const int end_shape[] = {1, 5}; in TF_LITE_MICRO_TEST() local
142 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
150 const int end_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
163 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
171 const int end_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
184 input_shape, begin_shape, end_shape, strides_shape, &builtin_data, in TF_LITE_MICRO_TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 35 std::initializer_list<int> end_shape, in StridedSliceOpModel() argument
49 BuildInterpreter({input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1706 ShapeHandle begin_shape, end_shape, strides_shape; in __anondb9326b22502() local
1708 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &end_shape)); in __anondb9326b22502()
1710 TF_RETURN_IF_ERROR(c->Merge(begin_shape, end_shape, &begin_shape)); in __anondb9326b22502()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2513 std::initializer_list<int> end_shape, in StridedSliceOpModel() argument
2521 end_ = AddConstInput(TensorType_INT32, end_data, end_shape); in StridedSliceOpModel()
2530 {input_shape, begin_shape, end_shape, strides_shape}); in StridedSliceOpModel()
|