Home
last modified time | relevance | path

Searched refs:compute_space_to_batch_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNESpaceToBatchLayerKernel.cpp78 …TensorShape expected_output_shape = misc::shape_calculator::compute_space_to_batch_shape(input, bl… in validate_arguments_static()
114 …TensorShape output_shape = misc::shape_calculator::compute_space_to_batch_shape(input->info(), blo… in configure()
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLSpaceToBatchLayerKernel.cpp74 …TensorShape expected_output_shape = misc::shape_calculator::compute_space_to_batch_shape(input, bl… in validate_arguments_static()
140 …TensorShape output_shape = misc::shape_calculator::compute_space_to_batch_shape(input->info(), blo… in configure()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h1165 inline TensorShape compute_space_to_batch_shape(const ITensorInfo *input, const int block_x, const … in compute_space_to_batch_shape() function