Searched defs:block_shape_data (Results 1 – 8 of 8) sorted by relevance
| /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 | 477 Tensor block_shape_data(DT_INT32, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1297 const auto& block_shape_data = in ProcessSpaceToBatchNDOperator() local
1346 const auto& block_shape_data = in ProcessBatchToSpaceNDOperator() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1920 const int32* block_shape_data, in SpaceToBatchND()
1937 const int32* block_shape_data, in SpaceToBatchND()
1953 const int32* block_shape_data, in BatchToSpaceND()
|
| D | reference_ops.h | 3078 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in SpaceToBatchND()
3136 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1785 const int32* block_shape_data, in BatchToSpaceND()
|
| D | optimized_ops.h | 5781 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6109 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND()
|