Home
last modified time | relevance | path

Searched refs:compute_pool3d_shape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuPool3dKernel.cpp110 …TensorInfo out_info(TensorInfo(compute_pool3d_shape(src->tensor_shape(), pool_info), 1, dst->data_… in validate_arguments()
129 …auto_init_if_empty(*dst, src->clone()->set_tensor_shape(compute_pool3d_shape(src->tensor_shape(), … in configure()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClPool3dKernel.cpp83 …TensorInfo out_info(TensorInfo(compute_pool3d_shape(src->tensor_shape(), pool_info), 1, dst->data_… in validate_arguments()
103 TensorShape out_shape = compute_pool3d_shape(src->tensor_shape(), pool_info); in configure()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DPooling3dLayerFixture.h80 …const TensorShape dst_shape = misc::shape_calculator::compute_pool3d_shape((src.info()->tensor_sha… in compute_target()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DPooling3dLayer.cpp42 TensorShape pooled_shape = compute_pool3d_shape(src.shape(), pool3d_info); in pooling_3d_layer_internal()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h1499 inline TensorShape compute_pool3d_shape(const TensorShape &src, Pooling3dLayerInfo pool3d_info) in compute_pool3d_shape() function