Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEBatchToSpaceLayerKernel.cpp106 …TensorShape output_shape = compute_batch_to_space_shape(input->info(), block_shape_x, block_shape_… in configure()
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLBatchToSpaceLayerKernel.cpp131 …TensorShape output_shape = compute_batch_to_space_shape(input->info(), block_shape_x, block_shape_… in configure()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h1081 inline TensorShape compute_batch_to_space_shape(const ITensorInfo *input, const int block_x, const … in compute_batch_to_space_shape() function