Searched refs:begin_data (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice_test.cc | 62 float* input_data, const int32_t* begin_data, in TestStridedSliceFloat() argument
77 CreateTensor(begin_data, begin_dims), in TestStridedSliceFloat()
93 const T* input_data, const int32_t* begin_data, const int32_t* end_data, in TestStridedSliceQuantized() argument
108 CreateTensor(begin_data, begin_dims), in TestStridedSliceQuantized()
133 int32_t begin_data[] = {}; in TF_LITE_MICRO_TEST() local
143 input_data, begin_data, end_data, strides_data, output_shape, output_data, in TF_LITE_MICRO_TEST()
154 int32_t begin_data[] = {1}; in TF_LITE_MICRO_TEST() local
164 input_data, begin_data, end_data, strides_data, output_shape, output_data, in TF_LITE_MICRO_TEST()
175 int32_t begin_data[] = {10}; in TF_LITE_MICRO_TEST() local
185 input_data, begin_data, end_data, strides_data, output_shape, output_data, in TF_LITE_MICRO_TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | slice_test.cc | 41 std::initializer_list<index_type> begin_data, in SliceOpModel() argument
52 AddConstInput(GetTensorType<index_type>(), begin_data, begin_shape); in SliceOpModel()
61 PopulateTensor<index_type>(begin_, begin_data); in SliceOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | sample_distorted_bounding_box_op.cc | 405 typename TTypes<T, 1>::Tensor begin_data(begin->tensor<T, 1>()); in DoCompute() local
409 begin_data(0) = T(offset_height); in DoCompute()
412 begin_data(1) = T(offset_width); in DoCompute()
425 begin_data(2) = T(0); in DoCompute()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | strided_slice_test.cc | 26 std::initializer_list<int> begin_data, in StridedSliceOpModel() argument
33 begin_ = AddConstInput(begin, begin_data); in StridedSliceOpModel()
|
| D | slice_test.cc | 26 std::initializer_list<index_type> begin_data, in SliceOpModel() argument
29 begin_ = AddConstInput(begin, begin_data); in SliceOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2512 std::initializer_list<int> begin_data, in StridedSliceOpModel() argument
2520 begin_ = AddConstInput(TensorType_INT32, begin_data, begin_shape); in StridedSliceOpModel()
|