| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_to_space_nd.h | 44 const int32_t* crops_data, in BatchToSpaceND() argument
70 const int crops_top = crops_data[0]; in BatchToSpaceND()
72 unextended_input1_shape.DimensionsCount() == 4 ? crops_data[2] : 0; in BatchToSpaceND()
|
| D | legacy_reference_ops.h | 2047 const int32* crops_data, const Dims<4>& crops_dims, in BatchToSpaceND() argument
2051 DimsToShape(crops_dims), crops_data, DimsToShape(output_dims), in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | batch_to_space_nd_test.cc | 69 const int* crops_dims_data, const int32_t* crops_data, in TestBatchToSpaceNdFloat() argument
82 CreateTensor(crops_data, crops_dims), in TestBatchToSpaceNdFloat()
95 const int32_t* crops_data, const int* output_dims_data, const float* golden, in TestBatchToSpaceNdQuantized() argument
111 tflite::testing::CreateTensor(crops_data, crops_dims), in TestBatchToSpaceNdQuantized()
|
| D | space_to_batch_nd_test.cc | 69 const int* crops_dims_data, const int32_t* crops_data, in TestSpaceToBatchNdFloat() argument
82 CreateTensor(crops_data, crops_dims), in TestSpaceToBatchNdFloat()
95 const int32_t* crops_data, const int* output_dims_data, const float* golden, in TestSpaceToBatchNdQuantized() argument
111 tflite::testing::CreateTensor(crops_data, crops_dims), in TestSpaceToBatchNdQuantized()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | flatten_atrous_test.cc | 73 Tensor crops_data(DT_INT32, TensorShape({2, 2})); in TestFlattenAtrousConv() local
74 test::FillValues<int>(&crops_data, {0, 1, 0, 1}); in TestFlattenAtrousConv()
76 Const(root.WithOpName("crops_op"), Input::Initializer(crops_data)); in TestFlattenAtrousConv()
|
| D | fold_old_batch_norms_test.cc | 483 Tensor crops_data(DT_INT32, TensorShape({2, 2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
484 test::FillValues<int32>(&crops_data, {0, 0, 0, 1}); in TestFoldFusedBatchNormsWithBatchToSpace()
486 Const(root.WithOpName("crops_op"), Input::Initializer(crops_data)); in TestFoldFusedBatchNormsWithBatchToSpace()
490 block_shape_op, crops_data); in TestFoldFusedBatchNormsWithBatchToSpace()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 2208 auto crops_data = crops.data.i32; in Validate() local
2209 Expect(crops_data && crops.bytes == 16 && crops_data[0] == 0 && in Validate()
2210 crops_data[1] == 0 && crops_data[2] == 0 && crops_data[3] == 0, in Validate()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1372 const auto& crops_data = crops_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessBatchToSpaceNDOperator() local
1383 crops_data[dim * 2] - crops_data[dim * 2 + 1]; in ProcessBatchToSpaceNDOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4989 const RuntimeShape& unextended_input3_shape, const int32* crops_data, in BatchToSpaceND() argument
5024 const int crops_top = crops_data[0]; in BatchToSpaceND()
5026 unextended_input1_shape.DimensionsCount() == 4 ? crops_data[2] : 0; in BatchToSpaceND()
|
| D | legacy_optimized_ops.h | 4845 const int32* crops_data, const Dims<4>& crops_dims, in BatchToSpaceND() argument
4849 DimsToShape(crops_dims), crops_data, DimsToShape(output_dims), in BatchToSpaceND()
|