| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLDepthToSpaceLayer.cpp | 33 void CLDepthToSpaceLayer::configure(const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
38 … CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
45 …pthToSpaceLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| D | CLSpaceToDepthLayer.cpp | 44 void CLSpaceToDepthLayer::configure(const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
49 … CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
54 …aceToDepthLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| D | CLBatchToSpaceLayer.cpp | 45 void CLBatchToSpaceLayer::configure(const ICLTensor *input, const ICLTensor *block_shape, ICLTensor… in configure()
50 …st CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *block_shape, ICLTen… in configure()
65 Status CLBatchToSpaceLayer::validate(const ITensorInfo *input, const ITensorInfo *block_shape, cons… in validate()
|
| 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()
82 Status CLSpaceToBatchLayer::validate(const ITensorInfo *input, const ITensorInfo *block_shape, cons… in validate()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLDepthToSpaceLayerKernel.cpp | 39 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate_arguments()
70 …LDepthToSpaceLayerKernel::configure(const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
75 … CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, int32_t block_shape) in configure()
108 …paceLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| D | CLSpaceToDepthLayerKernel.cpp | 39 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate_arguments()
71 …LSpaceToDepthLayerKernel::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()
105 …epthLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| D | CLBatchToSpaceLayerKernel.cpp | 88 void CLBatchToSpaceLayerKernel::configure(const ICLTensor *input, const ICLTensor *block_shape, ICL… in configure()
93 …st CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *block_shape, ICLTen… in configure()
154 Status CLBatchToSpaceLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *block_shape… in validate()
|
| D | CLSpaceToBatchLayerKernel.cpp | 89 void CLSpaceToBatchLayerKernel::configure(const ICLTensor *input, const ICLTensor *block_shape, con… in configure()
94 …st CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *block_shape, const … in configure()
172 Status CLSpaceToBatchLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *block_shape… in validate()
|
| /external/ComputeLibrary/src/graph/nodes/ |
| 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()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | DepthToSpaceLayerFixture.h | 43 …void setup(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, DataType data_t… in setup()
56 … TensorType compute_target(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, in compute_target()
92 SimpleTensor<T> compute_reference(const TensorShape &input_shape, int32_t block_shape, in compute_reference()
|
| D | SpaceToDepthFixture.h | 43 …void setup(TensorShape input_shape, TensorShape output_shape, const int block_shape, DataType data… in setup()
56 …TensorType compute_target(TensorShape input_shape, TensorShape output_shape, const int block_shape, in compute_target()
92 const int block_shape, DataType data_type) in compute_reference()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | spacetobatch_op_test.py | 34 def space_to_batch_direct(input_array, block_shape, paddings): argument
200 def _testPad(self, inputs, block_shape, paddings, outputs): argument
214 def _testDirect(self, input_shape, block_shape, paddings): argument
422 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
431 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
449 def _testShape(self, input_shape, block_shape, paddings, error): argument
586 def _checkGrad(self, x, block_shape, paddings): argument
603 def _compare(self, input_shape, block_shape, paddings): argument
629 def _checkProperties(self, input_shape, block_shape, base_paddings, paddings, argument
649 def _test(self, input_shape, block_shape, base_paddings): argument
|
| D | batchtospace_op_test.py | 139 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
148 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
167 def _testShape(self, input_shape, block_shape, paddings, error): argument
310 def _checkGrad(self, x, block_shape, crops, crops_dtype): argument
328 def _compare(self, input_shape, block_shape, crops, crops_dtype): argument
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEDepthToSpaceLayer.cpp | 37 void NEDepthToSpaceLayer::configure(const ITensor *input, ITensor *output, int32_t block_shape) in configure()
44 …pthToSpaceLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| D | NESpaceToDepthLayer.cpp | 44 void NESpaceToDepthLayer::configure(const ITensor *input, ITensor *output, int32_t block_shape) in configure()
51 …aceToDepthLayer::validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape) in validate()
|
| D | NEBatchToSpaceLayer.cpp | 37 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()
|
| D | NESpaceToBatchLayer.cpp | 45 void NESpaceToBatchLayer::configure(const ITensor *input, const ITensor *block_shape, const ITensor… in configure()
73 Status NESpaceToBatchLayer::validate(const ITensorInfo *input, const ITensorInfo *block_shape, cons… in validate()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| 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()
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | spacetobatch_op_test.py | 30 def space_to_batch_direct(input_array, block_shape, paddings): argument
155 def _testPad(self, inputs, block_shape, paddings, outputs): argument
187 def _testDirect(self, input_shape, block_shape, paddings): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1436 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1463 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1470 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1480 Tensor block_shape = test::AsTensor<int32>({3, 3}); in TEST() local
1490 Tensor block_shape = test::AsTensor<int32>({}); in TEST() local
1563 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
1594 Tensor block_shape = test::AsTensor<int32>({0, 2}); in TEST() local
1601 Tensor block_shape = test::AsTensor<int32>({1, 1}); in TEST() local
1612 Tensor block_shape = test::AsTensor<int32>({2, 2}); in TEST() local
1623 Tensor block_shape = test::AsTensor<int32>({2, 3}); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 26 absl::Span<const int64> block_shape, in BatchToSpace()
151 std::vector<int64> block_shape; in Compile() local
|
| D | spacetobatch_op.cc | 26 absl::Span<const int64> block_shape, in SpaceToBatch()
153 std::vector<int64> block_shape; in Compile() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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
|
| 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
|