Searched refs:compute_pool3d_shape (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
H A D | CpuPool3dKernel.cpp | 110 …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 D | ClPool3dKernel.cpp | 83 …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 D | Pooling3dLayerFixture.h | 80 …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 D | Pooling3dLayer.cpp | 42 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 D | ShapeCalculator.h | 1499 inline TensorShape compute_pool3d_shape(const TensorShape &src, Pooling3dLayerInfo pool3d_info) in compute_pool3d_shape() function
|