Home
last modified time | relevance | path

Searched refs:conv3d_info (Results 1 – 13 of 13) sorted by relevance

/external/ComputeLibrary/src/gpu/cl/kernels/
DClDirectConv3dKernel.cpp41 … ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const Conv3dInfo &conv3d_info) in validate_arguments() argument
47 … ARM_COMPUTE_RETURN_ERROR_ON_MSG(conv3d_info.act_info.enabled(), "Fused activation not supported"); in validate_arguments()
52 ARM_COMPUTE_RETURN_ERROR_ON(conv3d_info.dilation != Size3D(1U, 1U, 1U)); in validate_arguments()
57 …TURN_ERROR_ON(src1->dimension(2) > (src0->dimension(1) + conv3d_info.padding.left + conv3d_info.pa… in validate_arguments()
58 …ETURN_ERROR_ON(src1->dimension(3) > (src0->dimension(2) + conv3d_info.padding.top + conv3d_info.pa… in validate_arguments()
59 …TURN_ERROR_ON(src1->dimension(4) > (src0->dimension(3) + conv3d_info.padding.front + conv3d_info.p… in validate_arguments()
79 …::shape_calculator::compute_conv3d_shape(src0->tensor_shape(), src1->tensor_shape(), conv3d_info)); in validate_arguments()
93 const Conv3dInfo &conv3d_info) in configure() argument
98 ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(src0, src1, src2, dst, conv3d_info)); in configure()
113 const size_t pad_left = conv3d_info.padding.left; in configure()
[all …]
DClDirectConv3dKernel.h73 …const ITensorInfo *src1, const ITensorInfo *src2, ITensorInfo *dst, const Conv3dInfo &conv3d_info);
80 …ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const Conv3dInfo &conv3d_info);
/external/ComputeLibrary/src/runtime/CL/functions/
DCLConv3D.cpp49 …, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *dst, const Conv3dInfo &conv3d_info) in configure() argument
51 configure(CLKernelLibrary::get().get_compile_context(), src, weights, biases, dst, conv3d_info); in configure()
54 …, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *dst, const Conv3dInfo &conv3d_info) in configure() argument
57 …o(), weights->info(), ((biases != nullptr) ? biases->info() : nullptr), dst->info(), conv3d_info)); in configure()
65 …weights->info(), _impl->biases ? _impl->biases->info() : nullptr, _impl->dst->info(), conv3d_info); in configure()
68 …sorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv3dInfo &conv3d_info) in validate() argument
70 return opencl::ClDirectConv3d::validate(src, weights, biases, dst, conv3d_info); in validate()
/external/ComputeLibrary/tests/validation/reference/
DConv3D.cpp193 …sor<T> &weights, const SimpleTensor<TB> &bias, SimpleTensor<T> &dst, const Conv3dInfo &conv3d_info) in conv3d() argument
203 const size_t pad_left = conv3d_info.padding.left; in conv3d()
204 const size_t pad_top = conv3d_info.padding.top; in conv3d()
205 const size_t pad_front = conv3d_info.padding.front; in conv3d()
206 const size_t stride_x = conv3d_info.stride.x(); in conv3d()
207 const size_t stride_y = conv3d_info.stride.y(); in conv3d()
208 const size_t stride_z = conv3d_info.stride.z(); in conv3d()
210 …m_compute::misc::shape_calculator::compute_conv3d_shape(src.shape(), weights.shape(), conv3d_info); in conv3d()
240 …out_ptr[out_offset] = calculate_conv3d<T, TB>(src, weights, bias, conv3d_info.dilation, batch, z_… in conv3d()
250 const Conv3dInfo &conv3d_info);
[all …]
DConv3D.h42 const Conv3dInfo &conv3d_info);
/external/ComputeLibrary/src/gpu/cl/operators/
DClDirectConv3d.cpp33 … const ITensorInfo *src1, const ITensorInfo *src2, ITensorInfo *dst, const Conv3dInfo &conv3d_info) in configure() argument
39 k->configure(compile_context, src0, src1, src2, dst, conv3d_info); in configure()
43 … ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const Conv3dInfo &conv3d_info) in validate() argument
45 …MPUTE_RETURN_ON_ERROR(kernels::ClDirectConv3dKernel::validate(src0, src1, src2, dst, conv3d_info)); in validate()
DClDirectConv3d.h70 …const ITensorInfo *src1, const ITensorInfo *src2, ITensorInfo *dst, const Conv3dInfo &conv3d_info);
78 …ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const Conv3dInfo &conv3d_info);
/external/ComputeLibrary/tests/validation/fixtures/
DDirectConvolution3DFixture.h55 …const Conv3dInfo conv3d_info(Size3D(stride_x, stride_y, stride_z), Padding3D(pad_x, pad_y, pad_z)…
56 … const TensorShape output_shape = compute_conv3d_shape(input_shape, weights_shape, conv3d_info);
58 …= compute_target(input_shape, weights_shape, bias_shape, output_shape, conv3d_info, has_bias, data…
59 …ompute_reference(input_shape, weights_shape, bias_shape, output_shape, conv3d_info, has_bias, data…
85 …ape, const TensorShape &bias_shape, const TensorShape &output_shape, const Conv3dInfo &conv3d_info, in compute_target() argument
97 conv.configure(&src, &weights, has_bias ? &bias : nullptr, &dst, conv3d_info); in compute_target()
131 …const Conv3dInfo &conv3d_info, bool has_bias, const DataType &data_type, const DataType &bias_data… in compute_reference() argument
149 …tion_layer(reference::conv3d<T, TBias>(src, weights, bias, dst, conv3d_info), conv3d_info.act_info… in compute_reference()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLConv3D.h80 … const ICLTensor *weights, const ICLTensor *biases, ICLTensor *dst, const Conv3dInfo &conv3d_info);
86 … const ICLTensor *weights, const ICLTensor *biases, ICLTensor *dst, const Conv3dInfo &conv3d_info);
93 …orInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv3dInfo &conv3d_info);
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h1435 …ute_conv3d_shape(const TensorShape &src, const TensorShape &weights, const Conv3dInfo &conv3d_info) in compute_conv3d_shape() argument
1451 const size_t pad_left = conv3d_info.padding.left; in compute_conv3d_shape()
1452 const size_t pad_right = conv3d_info.padding.right; in compute_conv3d_shape()
1453 const size_t pad_top = conv3d_info.padding.top; in compute_conv3d_shape()
1454 const size_t pad_bottom = conv3d_info.padding.bottom; in compute_conv3d_shape()
1455 const size_t pad_front = conv3d_info.padding.front; in compute_conv3d_shape()
1456 const size_t pad_back = conv3d_info.padding.back; in compute_conv3d_shape()
1457 const size_t dilation_x = conv3d_info.dilation.width; in compute_conv3d_shape()
1458 const size_t dilation_y = conv3d_info.dilation.height; in compute_conv3d_shape()
1459 const size_t dilation_z = conv3d_info.dilation.depth; in compute_conv3d_shape()
[all …]
/external/ComputeLibrary/tests/validation/NEON/
DConvolution3D.cpp122 …const Conv3dInfo conv3d_info(Size3D(1, 1, 1), Padding3D(0, 0, 0), ActivationLayerInfo(), Size3D(1… local
123 …nfo.clone()->set_is_resizable(false), &output_info.clone()->set_is_resizable(false), conv3d_info));
/external/ComputeLibrary/utils/
DTypePrinter.h3267 inline ::std::ostream &operator<<(::std::ostream &os, const Conv3dInfo &conv3d_info)
3269 os << conv3d_info.stride;
3271 os << conv3d_info.padding;
3273 os << to_string(conv3d_info.act_info);
3275 os << conv3d_info.dilation;
3277 os << conv3d_info.round_type;
3279 os << conv3d_info.enable_fast_math;
3290 inline std::string to_string(const Conv3dInfo &conv3d_info) in to_string() argument
3293 str << conv3d_info; in to_string()
/external/ComputeLibrary/tests/validation/CL/
DConvolution3D.cpp152 …input_shape, weights_shape, biases_shape, output_shape, conv3d_info, src_data_type, weights_data_t…
164 …nfo.clone()->set_is_resizable(false), &output_info.clone()->set_is_resizable(false), conv3d_info));