Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/cpu/kernels/
DCpuDirectConv3dKernel.cpp116 …TensorShape output_shape = misc::shape_calculator::compute_conv3d_shape(src0->tensor_shape(), src1… in validate_arguments()
142 …TensorShape output_shape = misc::shape_calculator::compute_conv3d_shape(src0->tensor_shape(), src1… in configure()
/external/ComputeLibrary/tests/validation/fixtures/
DDirectConvolution3DFixture.h56 … const TensorShape output_shape = compute_conv3d_shape(input_shape, weights_shape, conv3d_info);
/external/ComputeLibrary/tests/validation/reference/
DConv3D.cpp210 …const TensorShape dst_shape = arm_compute::misc::shape_calculator::compute_conv3d_shape(src.shape(… in conv3d()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClDirectConv3dKernel.cpp79 …SMATCHING_DIMENSIONS(dst->tensor_shape(), misc::shape_calculator::compute_conv3d_shape(src0->tenso… in validate_arguments()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h1435 inline TensorShape compute_conv3d_shape(const TensorShape &src, const TensorShape &weights, const C… in compute_conv3d_shape() function