Home
last modified time | relevance | path

Searched refs:block_shape_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dflatten_atrous_test.cc50 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()
Dfold_old_batch_norms_test.cc477 Tensor block_shape_data(DT_INT32, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
478 test::FillValues<int32>(&block_shape_data, {1, 2}); in TestFoldFusedBatchNormsWithBatchToSpace()
480 Input::Initializer(block_shape_data)); in TestFoldFusedBatchNormsWithBatchToSpace()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1297 const auto& block_shape_data = in ProcessSpaceToBatchNDOperator() local
1299 auto block_height = block_shape_data[0]; in ProcessSpaceToBatchNDOperator()
1300 auto block_width = block_shape_data[1]; in ProcessSpaceToBatchNDOperator()
1346 const auto& block_shape_data = in ProcessBatchToSpaceNDOperator() local
1348 auto block_height = block_shape_data[0]; in ProcessBatchToSpaceNDOperator()
1349 auto block_width = block_shape_data[1]; in ProcessBatchToSpaceNDOperator()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h1920 const int32* block_shape_data, in SpaceToBatchND() argument
1930 DimsToShape(block_shape_dims), block_shape_data, in SpaceToBatchND()
1937 const int32* block_shape_data, in SpaceToBatchND() argument
1946 DimsToShape(block_shape_dims), block_shape_data, in SpaceToBatchND()
1953 const int32* block_shape_data, in BatchToSpaceND() argument
1958 DimsToShape(block_shape_dims), block_shape_data, in BatchToSpaceND()
Dreference_ops.h3078 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in SpaceToBatchND() argument
3098 const int block_shape_height = block_shape_data[0]; in SpaceToBatchND()
3099 const int block_shape_width = block_shape_data[1]; in SpaceToBatchND()
3136 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND() argument
3156 const int block_shape_width = block_shape_data[1]; in BatchToSpaceND()
3157 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1785 const int32* block_shape_data, in BatchToSpaceND() argument
1790 DimsToShape(block_shape_dims), block_shape_data, in BatchToSpaceND()
Doptimized_ops.h5781 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND() argument
5802 const int block_shape_width = block_shape_data[1]; in BatchToSpaceND()
5803 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h6109 const RuntimeShape& unextended_input2_shape, const int32* block_shape_data, in BatchToSpaceND() argument
6130 const int block_shape_width = block_shape_data[1]; in BatchToSpaceND()
6131 const int block_shape_height = block_shape_data[0]; in BatchToSpaceND()