Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DDepthToSpaceLayerNode.cpp56 …output_descriptor.shape = misc::shape_calculator::compute_depth_to_space_shape(input_sh… in compute_output_descriptor()
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLDepthToSpaceLayerKernel.cpp80 …TensorShape output_shape = compute_depth_to_space_shape(input->info()->tensor_shape(), input->info… in configure()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEDepthToSpaceLayerKernel.cpp77 …TensorShape output_shape = compute_depth_to_space_shape(input->info()->tensor_shape(), input->info… in configure()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h1106 inline TensorShape compute_depth_to_space_shape(const TensorShape &input_shape, DataLayout data_lay… in compute_depth_to_space_shape() function