| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_to_space_nd.h | 42 const int32_t* block_shape_data, in BatchToSpaceND() argument
67 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND()
69 unextended_input1_shape.DimensionsCount() == 4 ? block_shape_data[1] : 1; in BatchToSpaceND()
|
| D | space_to_batch_nd.h | 44 const int32_t* block_shape_data, in SpaceToBatchND() argument
70 const int block_shape_height = block_shape_data[0]; in SpaceToBatchND()
72 unextended_input1_shape.DimensionsCount() == 4 ? block_shape_data[1] : 1; in SpaceToBatchND()
|
| D | legacy_reference_ops.h | 2012 const int32* block_shape_data, in SpaceToBatchND() argument
2022 DimsToShape(block_shape_dims), block_shape_data, in SpaceToBatchND()
2029 const int32* block_shape_data, in SpaceToBatchND() argument
2038 DimsToShape(block_shape_dims), block_shape_data, in SpaceToBatchND()
2045 const int32* block_shape_data, in BatchToSpaceND() argument
2050 DimsToShape(block_shape_dims), block_shape_data, in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | batch_to_space_nd_test.cc | 68 const int* block_shape_dims_data, const int32_t* block_shape_data, in TestBatchToSpaceNdFloat() argument
81 CreateTensor(block_shape_data, block_shape_dims), in TestBatchToSpaceNdFloat()
94 const int32_t* block_shape_data, const int* crops_dims_data, in TestBatchToSpaceNdQuantized() argument
110 tflite::testing::CreateTensor(block_shape_data, block_shape_dims), in TestBatchToSpaceNdQuantized()
|
| D | space_to_batch_nd_test.cc | 68 const int* block_shape_dims_data, const int32_t* block_shape_data, in TestSpaceToBatchNdFloat() argument
81 CreateTensor(block_shape_data, block_shape_dims), in TestSpaceToBatchNdFloat()
94 const int32_t* block_shape_data, const int* crops_dims_data, in TestSpaceToBatchNdQuantized() argument
110 tflite::testing::CreateTensor(block_shape_data, block_shape_dims), in TestSpaceToBatchNdQuantized()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | flatten_atrous_test.cc | 50 Tensor block_shape_data(DT_INT32, TensorShape({2})); in TestFlattenAtrousConv() local
51 test::FillValues<int>(&block_shape_data, {2, 2}); in TestFlattenAtrousConv()
53 Input::Initializer(block_shape_data)); in TestFlattenAtrousConv()
|
| D | fold_old_batch_norms_test.cc | 478 Tensor block_shape_data(DT_INT32, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
479 test::FillValues<int32>(&block_shape_data, {1, 2}); in TestFoldFusedBatchNormsWithBatchToSpace()
481 Input::Initializer(block_shape_data)); in TestFoldFusedBatchNormsWithBatchToSpace()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1309 const auto& block_shape_data = in ProcessSpaceToBatchNDOperator() local
1327 QCHECK_EQ(final_dim_size % block_shape_data[dim], 0); in ProcessSpaceToBatchNDOperator()
1328 output_shape_data->at(dim + 1) = final_dim_size / block_shape_data[dim]; in ProcessSpaceToBatchNDOperator()
1329 output_batch_size *= block_shape_data[dim]; in ProcessSpaceToBatchNDOperator()
1363 const auto& block_shape_data = in ProcessBatchToSpaceNDOperator() local
1379 QCHECK_EQ(output_batch_size % block_shape_data[dim], 0); in ProcessBatchToSpaceNDOperator()
1380 output_batch_size = output_batch_size / block_shape_data[dim]; in ProcessBatchToSpaceNDOperator()
1382 input_shape.dims(dim + 1) * block_shape_data[dim] - in ProcessBatchToSpaceNDOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4988 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND() argument
5021 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND()
5023 unextended_input1_shape.DimensionsCount() == 4 ? block_shape_data[1] : 1; in BatchToSpaceND()
|
| D | legacy_optimized_ops.h | 4843 const int32* block_shape_data, in BatchToSpaceND() argument
4848 DimsToShape(block_shape_dims), block_shape_data, in BatchToSpaceND()
|