Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A 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()
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLReorgLayerKernel.cpp59 …ut = output->clone()->set_tensor_shape(misc::shape_calculator::compute_reorg_output_shape(*input, … in validate_arguments()
100 …put->info()->clone()->set_tensor_shape(misc::shape_calculator::compute_reorg_output_shape(*input->… in configure()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DReorgLayer.cpp44 …const TensorShape output_shape = misc::shape_calculator::compute_reorg_output_shape(input_info, st… in reorg_layer()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h123 inline TensorShape compute_reorg_output_shape(const ITensorInfo &input, int32_t stride) in compute_reorg_output_shape() function