Searched defs:block_shape_data (Results 1 – 10 of 10) sorted by relevance
| /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()
94 const int32_t* block_shape_data, const int* crops_dims_data, in TestBatchToSpaceNdQuantized()
|
| D | space_to_batch_nd_test.cc | 68 const int* block_shape_dims_data, const int32_t* block_shape_data, in TestSpaceToBatchNdFloat()
94 const int32_t* block_shape_data, const int* crops_dims_data, 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
|
| D | fold_old_batch_norms_test.cc | 478 Tensor block_shape_data(DT_INT32, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_to_space_nd.h | 42 const int32_t* block_shape_data, in BatchToSpaceND()
|
| D | space_to_batch_nd.h | 44 const int32_t* block_shape_data, in SpaceToBatchND()
|
| D | legacy_reference_ops.h | 2012 const int32* block_shape_data, in SpaceToBatchND()
2029 const int32* block_shape_data, in SpaceToBatchND()
2045 const int32* block_shape_data, in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1309 const auto& block_shape_data = in ProcessSpaceToBatchNDOperator() local
1363 const auto& block_shape_data = in ProcessBatchToSpaceNDOperator() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4843 const int32* block_shape_data, in BatchToSpaceND()
|
| D | optimized_ops.h | 4988 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND()
|