Searched defs:block_shape_data (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | BatchToSpaceLayerFixture.h | 93 auto block_shape_data = AccessorType(block_shape); in compute_target() local
97 …static_cast<int32_t *>(block_shape_data.data())[i] = output_shape[i + idx_width] / input_shape[i +… in compute_target() local
|
| H A D | SpaceToBatchFixture.h | 102 auto block_shape_data = AccessorType(block_shape); in compute_target() local
106 …static_cast<int32_t *>(block_shape_data.data())[i] = input_shape[i + idx_width] / output_shape[i +… in compute_target() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | flatten_atrous_test.cc | 50 Tensor block_shape_data(DT_INT32, TensorShape({2})); in TestFlattenAtrousConv() local
|
| H A D | fold_old_batch_norms_test.cc | 478 Tensor block_shape_data(DT_INT32, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | batch_to_space_nd.h | 42 const int32_t* block_shape_data, in BatchToSpaceND()
|
| H A D | space_to_batch_nd.h | 44 const int32_t* block_shape_data, in SpaceToBatchND()
|
| H A D | legacy_reference_ops.h | 2019 const int32* block_shape_data, in SpaceToBatchND()
2036 const int32* block_shape_data, in SpaceToBatchND()
2052 const int32* block_shape_data, in BatchToSpaceND()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | propagate_fixed_sizes.cc | 1309 const auto& block_shape_data = in ProcessSpaceToBatchNDOperator() local
1363 const auto& block_shape_data = in ProcessBatchToSpaceNDOperator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 4848 const int32* block_shape_data, in BatchToSpaceND()
|
| H A D | optimized_ops.h | 4199 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND()
|