| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | spacetobatch_op_test.py | 32 def space_to_batch_direct(input_array, block_shape, paddings): argument
198 def _testPad(self, inputs, block_shape, paddings, outputs): argument
212 def _testDirect(self, input_shape, block_shape, paddings): argument
420 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
429 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
447 def _testShape(self, input_shape, block_shape, paddings, error): argument
605 def _checkGrad(self, x, block_shape, paddings): argument
622 def _compare(self, input_shape, block_shape, paddings): argument
648 def _checkProperties(self, input_shape, block_shape, base_paddings, paddings, argument
668 def _test(self, input_shape, block_shape, base_paddings): argument
|
| H A D | batchtospace_op_test.py | 135 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
144 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
163 def _testShape(self, input_shape, block_shape, paddings, error): argument
306 def _checkGrad(self, x, block_shape, crops, crops_dtype): argument
324 def _compare(self, input_shape, block_shape, crops, crops_dtype): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops_test.cc | 1438 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1465 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1472 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1482 Tensor block_shape = test::AsTensor<int32>({3, 3}); in TEST() local
1492 Tensor block_shape = test::AsTensor<int32>({}); in TEST() local
1565 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1596 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1603 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1614 Tensor block_shape = test::AsTensor<int32>({2, 2}); in TEST() local
1625 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLDepthToSpaceLayerKernel.cpp | 39 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate_arguments()
71 …LDepthToSpaceLayerKernel::configure(const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
76 … CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
109 …paceLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| H A D | CLSpaceToDepthLayerKernel.cpp | 39 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate_arguments()
72 …LSpaceToDepthLayerKernel::configure(const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
77 … CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
108 …epthLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| H A D | CLBatchToSpaceLayerKernel.cpp | 89 void CLBatchToSpaceLayerKernel::configure(const ICLTensor *input, const ICLTensor *block_shape, ICL… in configure()
94 …st CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *block_shape, ICLTen… in configure()
155 Status CLBatchToSpaceLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *block_shape… in validate()
|
| H A D | CLSpaceToBatchLayerKernel.cpp | 90 void CLSpaceToBatchLayerKernel::configure(const ICLTensor *input, const ICLTensor *block_shape, con… in configure()
95 …st CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *block_shape, const … in configure()
175 Status CLSpaceToBatchLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *block_shape… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLDepthToSpaceLayer.cpp | 34 void CLDepthToSpaceLayer::configure(const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
39 … CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
47 …pthToSpaceLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| H A D | CLSpaceToDepthLayer.cpp | 45 void CLSpaceToDepthLayer::configure(const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
50 … CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
56 …aceToDepthLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| H A D | CLBatchToSpaceLayer.cpp | 46 void CLBatchToSpaceLayer::configure(const ICLTensor *input, const ICLTensor *block_shape, ICLTensor… in configure()
51 …st CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *block_shape, ICLTen… in configure()
68 Status CLBatchToSpaceLayer::validate(const ITensorInfo *input, const ITensorInfo *block_shape, cons… in validate()
|
| H A D | CLSpaceToBatchLayer.cpp | 47 void CLSpaceToBatchLayer::configure(const ICLTensor *input, const ICLTensor *block_shape, const ICL… in configure()
52 …st CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *block_shape, const … in configure()
84 Status CLSpaceToBatchLayer::validate(const ITensorInfo *input, const ITensorInfo *block_shape, cons… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | DepthToSpaceLayerNode.cpp | 35 DepthToSpaceLayerNode::DepthToSpaceLayerNode(int block_shape) in DepthToSpaceLayerNode()
42 int DepthToSpaceLayerNode::block_shape() const in block_shape() function in arm_compute::graph::DepthToSpaceLayerNode
47 …paceLayerNode::compute_output_descriptor(const TensorDescriptor &input_descriptor, int block_shape) in compute_output_descriptor()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | DepthToSpaceLayerFixture.h | 43 …void setup(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, DataType data_t… in setup()
59 … TensorType compute_target(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, in compute_target()
95 SimpleTensor<T> compute_reference(const TensorShape &input_shape, int32_t block_shape, in compute_reference()
|
| H A D | SpaceToDepthFixture.h | 44 …void setup(TensorShape input_shape, TensorShape output_shape, const int block_shape, DataType data… in setup()
60 …TensorType compute_target(TensorShape input_shape, TensorShape output_shape, const int block_shape, in compute_target()
102 const int block_shape, DataType data_type) in compute_reference()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NESpaceToDepthLayerKernel.cpp | 44 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate_arguments()
77 …id NESpaceToDepthLayerKernel::configure(const ITensor *input, ITensor *output, int32_t block_shape) in configure()
96 …epthLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| H A D | NEDepthToSpaceLayerKernel.cpp | 44 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate_arguments()
74 …id NEDepthToSpaceLayerKernel::configure(const ITensor *input, ITensor *output, int32_t block_shape) in configure()
94 …paceLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | spacetobatch_op_test.py | 27 def space_to_batch_direct(input_array, block_shape, paddings): argument
175 def _testPad(self, inputs, block_shape, paddings, outputs): argument
207 def _testDirect(self, input_shape, block_shape, paddings): argument
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEDepthToSpaceLayer.cpp | 36 void NEDepthToSpaceLayer::configure(const ITensor *input, ITensor *output, int32_t block_shape) in configure()
45 …pthToSpaceLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| H A D | NESpaceToDepthLayer.cpp | 44 void NESpaceToDepthLayer::configure(const ITensor *input, ITensor *output, int32_t block_shape) in configure()
53 …aceToDepthLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| H A D | NEBatchToSpaceLayer.cpp | 36 void NEBatchToSpaceLayer::configure(const ITensor *input, const ITensor *block_shape, ITensor *outp… in configure()
51 Status NEBatchToSpaceLayer::validate(const ITensorInfo *input, const ITensorInfo *block_shape, cons… in validate()
|
| H A D | NESpaceToBatchLayer.cpp | 45 void NESpaceToBatchLayer::configure(const ITensor *input, const ITensor *block_shape, const ITensor… in configure()
74 Status NESpaceToBatchLayer::validate(const ITensorInfo *input, const ITensorInfo *block_shape, cons… in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | batchtospace_op.cc | 26 absl::Span<const int64_t> block_shape, in BatchToSpace()
151 std::vector<int64_t> block_shape; in Compile() local
|
| H A D | spacetobatch_op.cc | 27 absl::Span<const int64_t> block_shape, in SpaceToBatch()
167 std::vector<int64_t> block_shape; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | batch_to_space_nd.cc | 44 const TfLiteTensor* block_shape; member
59 const int* block_shape = GetTensorData<int32>(op_context->block_shape); in ResizeOutputTensor() local
|
| H A D | space_to_batch_nd.cc | 45 const TfLiteTensor* block_shape; member
60 const int32* block_shape = GetTensorData<int32>(op_context->block_shape); in ResizeOutputTensor() local
|