Home
last modified time | relevance | path

Searched refs:compute_reorg_output_shape (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/
DNEReorgLayerKernel.cpp59 …ut = output->clone()->set_tensor_shape(misc::shape_calculator::compute_reorg_output_shape(*input, … in validate_arguments()
78 …const TensorShape output_shape = misc::shape_calculator::compute_reorg_output_shape(*input->info()… in configure()
/external/ComputeLibrary/src/core/CL/kernels/
DCLReorgLayerKernel.cpp59 …ut = output->clone()->set_tensor_shape(misc::shape_calculator::compute_reorg_output_shape(*input, … in validate_arguments()
98 …put->info()->clone()->set_tensor_shape(misc::shape_calculator::compute_reorg_output_shape(*input->… in configure()
/external/ComputeLibrary/tests/validation/reference/
DReorgLayer.cpp44 …const TensorShape output_shape = misc::shape_calculator::compute_reorg_output_shape(input_info, st… in reorg_layer()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h122 inline TensorShape compute_reorg_output_shape(const ITensorInfo &input, int32_t stride) in compute_reorg_output_shape() function