Home
last modified time | relevance | path

Searched refs:weights_info (Results 1 – 25 of 57) sorted by relevance

123

/external/ComputeLibrary/src/core/CL/kernels/
DCLDeconvolutionReshapeOutputKernel.cpp41 …o *bias, const ITensorInfo *output, const ITensorInfo *input_info, const ITensorInfo *weights_info, in validate_arguments() argument
44 ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(input, output, input_info, weights_info); in validate_arguments()
53 ARM_COMPUTE_RETURN_ERROR_ON(weights_info->dimension(idx_w) != deconv_info.stride().first); in validate_arguments()
54 ARM_COMPUTE_RETURN_ERROR_ON(weights_info->dimension(idx_h) != deconv_info.stride().second); in validate_arguments()
59 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input, input_info, weights_info); in validate_arguments()
61 …TURN_ERROR_ON(input->dimension(0) != weights_info->dimension(idx_w) * weights_info->dimension(idx_… in validate_arguments()
76 ARM_COMPUTE_RETURN_ERROR_ON(bias->dimension(0) != weights_info->dimension(idx_b)); in validate_arguments()
82 …info->dimension(idx_w), input_info->dimension(idx_h), weights_info->dimension(idx_w), weights_info… in validate_arguments()
84 …= misc::shape_calculator::compute_deconvolution_output_shape(out_dims, *input_info, *weights_info); in validate_arguments()
91 …TensorInfo *output, const ITensorInfo *input_info, const ITensorInfo *weights_info, const PadStrid… in validate_and_configure_window() argument
[all …]
DCLDeconvolutionReshapeOutputKernel.h70 … ICLTensor *output, const ITensorInfo *input_info, const ITensorInfo *weights_info, const PadStrid…
82 …ICLTensor *bias, ICLTensor *output, const ITensorInfo *input_info, const ITensorInfo *weights_info,
96 …TensorInfo *output, const ITensorInfo *input_info, const ITensorInfo *weights_info, const PadStrid…
/external/ComputeLibrary/src/runtime/CL/functions/
DCLDeconvolutionLayer.cpp61 const WeightsInfo &weights_info) in configure() argument
63 …rnelLibrary::get().get_compile_context(), input, weights, bias, output, deconv_info, weights_info); in configure()
67 const WeightsInfo &weights_info) in configure() argument
70 ARM_COMPUTE_LOG_PARAMS(input, weights, bias, output, deconv_info, weights_info); in configure()
72 …olution_method(input->info(), weights->info(), nullptr, output->info(), deconv_info, weights_info)) in configure()
90 f->configure(compile_context, input, weights, bias, output, deconv_info, weights_info); in configure()
108 const WeightsInfo &weights_info) in validate() argument
111 …onvolutionLayer::get_deconvolution_method(input, weights, bias, output, deconv_info, weights_info)) in validate()
122 …ROR(CLDirectDeconvolutionLayer::validate(input, weights, bias, output, deconv_info, weights_info)); in validate()
140 const WeightsInfo &weights_info) in get_deconvolution_method() argument
[all …]
DCLConvolutionLayer.cpp63 …Tensor *biases, ICLTensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, in configure() argument
66 …t().get_compile_context(), input, weights, biases, output, conv_info, weights_info, dilation, act_… in configure()
70 const WeightsInfo &weights_info, in configure() argument
74 …s != nullptr) ? biases->info() : nullptr), output->info(), conv_info, weights_info, dilation, act_… in configure()
76 …ARM_COMPUTE_LOG_PARAMS(input, weights, biases, output, conv_info, weights_info, dilation, act_info… in configure()
86 weights_info, CLScheduler::get().target())) in configure()
94 …fo(), ((biases != nullptr) ? biases->info() : nullptr), output->info(), conv2d_info, weights_info); in configure()
130 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, bool… in validate() argument
138 …l::ClConv2d::get_convolution_method(input, weights, output, conv2d_info, weights_info, gpu_target)) in validate()
145 …RN_ON_ERROR(opencl::ClConv2d::validate(input, weights, biases, output, conv2d_info, weights_info)); in validate()
[all …]
DCLGEMMConvolutionLayer.cpp71 …Tensor *biases, ICLTensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, in configure() argument
74 …t().get_compile_context(), input, weights, biases, output, conv_info, weights_info, dilation, act_… in configure()
79 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, unsi… in configure() argument
90 …info(), (biases != nullptr ? biases->info() : nullptr), output->info(), conv2d_info, weights_info); in configure()
118 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, unsi… in validate() argument
121 … return opencl::ClGemmConv2d::validate(input, weights, biases, output, conv2d_info, weights_info); in validate()
DCLDirectDeconvolutionLayer.cpp59 const WeightsInfo &weights_info) in validate() argument
110 …ROR(CLConvolutionLayer::validate(&scale_out_info, weights, bias, output, conv_info, weights_info)); in validate()
116 const WeightsInfo &weights_info) in configure() argument
118 …re(CLKernelLibrary::get().get_compile_context(), input, weights, bias, output, info, weights_info); in configure()
122 const WeightsInfo &weights_info) in configure() argument
125 ARM_COMPUTE_LOG_PARAMS(input, weights, bias, output, info, weights_info); in configure()
154 _is_prepared = weights_info.retain_internal_weights(); in configure()
187 …figure(compile_context, &_scaled_output, &_weights_flipped, bias, output, conv_info, weights_info); in configure()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEConvolutionLayer.cpp62 … ITensor *biases, ITensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, in configure() argument
68 …s != nullptr) ? biases->info() : nullptr), output->info(), conv_info, weights_info, dilation, act_… in configure()
70 …ARM_COMPUTE_LOG_PARAMS(input, weights, biases, output, conv_info, weights_info, dilation, act_info… in configure()
73 …ion_method(input->info(), weights->info(), output->info(), conv_info, weights_info, dilation, act_… in configure()
81 …s != nullptr) ? biases->info() : nullptr), output->info(), conv_info, weights_info, dilation, act_… in configure()
108 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, bool… in validate() argument
111 …switch(cpu::CpuConv2d::get_convolution_method(input, weights, output, conv_info, weights_info, dil… in validate()
117 …R(cpu::CpuConv2d::validate(input, weights, biases, output, conv_info, weights_info, dilation, act_… in validate()
131 … const WeightsInfo &weights_info, const Size2D &dilation, in get_convolution_method() argument
134 …return cpu::CpuConv2d::get_convolution_method(input, weights, output, conv_info, weights_info, dil… in get_convolution_method()
DNEFullyConnectedLayer.cpp64 … FullyConnectedLayerInfo fc_info, const WeightsInfo &weights_info) in configure() argument
73 weights_info)); in configure()
80 …ts->info(), (biases != nullptr) ? biases->info() : nullptr, output->info(), fc_info, weights_info); in configure()
94 const WeightsInfo &weights_info) in has_opt_impl() argument
96 …ected::has_opt_impl(expected_weight_format, input, weights, biases, output, fc_info, weights_info); in has_opt_impl()
100 … FullyConnectedLayerInfo fc_info, const WeightsInfo &weights_info) in validate() argument
102 return cpu::CpuFullyConnected::validate(input, weights, biases, output, fc_info, weights_info); in validate()
DNEGEMMConvolutionLayer.cpp57 … ITensor *biases, ITensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, in configure() argument
64 …es != nullptr ? biases->info() : nullptr), output->info(), conv_info, weights_info, dilation, act_… in configure()
78 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, bool… in validate() argument
80 …return cpu::CpuGemmConv2d::validate(input, weights, biases, output, conv_info, weights_info, dilat… in validate()
85 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, cons… in has_opt_impl() argument
87 …pt_impl(expected_weight_format, src, weights, biases, dst, conv_info, weights_info, dilation, act_… in has_opt_impl()
/external/ComputeLibrary/src/cpu/operators/
DCpuConv2d.cpp45 …Info *biases, ITensorInfo *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, in configure() argument
51 …ROW_ON(CpuConv2d::validate(input, weights, biases, output, conv_info, weights_info, dilation, act_… in configure()
54 …ARM_COMPUTE_LOG_PARAMS(input, weights, biases, output, conv_info, weights_info, dilation, act_info… in configure()
57 …switch(CpuConv2d::get_convolution_method(input, weights, output, conv_info, weights_info, dilation… in configure()
69 …f->configure(input, weights, biases, output, conv_info, weights_info, dilation, act_info, enable_f… in configure()
96 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, bool… in validate() argument
101 …switch(CpuConv2d::get_convolution_method(input, weights, output, conv_info, weights_info, dilation… in validate()
107 …OR(CpuGemmConv2d::validate(input, weights, biases, output, conv_info, weights_info, dilation, act_… in validate()
125 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, bool… in get_convolution_method() argument
128 ARM_COMPUTE_UNUSED(weights_info); in get_convolution_method()
DCpuFullyConnected.cpp130 … TensorInfo weights_info = weights->clone()->set_quantization_info(weights_quantization_info); in validate_mm() local
132 &weights_info, in validate_mm()
185 … TensorInfo weights_info = weights->clone()->set_quantization_info(weights_quantization_info); in configure_mm() local
189 …const Status status = get_gemmlowp_output_stage_info(&src_info, &weights_info, dst, act… in configure_mm()
197 _mm_gemmlowp->configure(&src_info, &weights_info, biases, dst, gemm_info); in configure_mm()
237 FullyConnectedLayerInfo fc_info, const WeightsInfo &weights_info) in configure() argument
246 weights_info)); in configure()
257 _fixed_format = weights_info.weight_format() != WeightFormat::UNSPECIFIED; in configure()
258 _weight_format = weights_info.weight_format(); in configure()
346 …sorInfo *biases, const ITensorInfo *dst, FullyConnectedLayerInfo fc_info, WeightsInfo weights_info) in has_opt_impl() argument
[all …]
DCpuGemmConv2d.cpp242 …2d::validate_gemm3d(const ITensorInfo *input_info, const ITensorInfo *weights_info, const Activati… in validate_gemm3d() argument
250 …const TensorInfo dummy_weights_info(TensorShape(4U, 4U), 1, data_type, weights_info->quantization_… in validate_gemm3d()
256 …sorInfo *biases, ITensorInfo *dst, const PadStrideInfo &conv_info, const WeightsInfo &weights_info, in configure() argument
260 ARM_COMPUTE_UNUSED(num_groups, weights_info); in configure()
266 weights_info, in configure()
271 …ARM_COMPUTE_LOG_PARAMS(src, weights, biases, dst, conv_info, weights_info, dilation, act_info, ena… in configure()
282 _is_prepared = weights_info.retain_internal_weights(); in configure()
363 …const bool fixed_format = weights_info.weight_format() != arm_compute::WeightFormat::UNSP… in configure()
364 …tput_to_use, act_info, enable_fast_math, gemm_3d_depth, fixed_format, weights_info.weight_format()… in configure()
393 …const WeightsInfo &weights_info, const Size2D &dilation, const ActivationLayerInfo &act_info, cons… in has_opt_impl() argument
[all …]
DCpuConv2d.h104 … ITensorInfo *dst, const PadStrideInfo &conv_info, const WeightsInfo &weights_info = WeightsInfo(),
113 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
135 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
DCpuFullyConnected.h89 …ctedLayerInfo fc_info = FullyConnectedLayerInfo(), const WeightsInfo &weights_info = WeightsInfo()…
97 …ctedLayerInfo fc_info = FullyConnectedLayerInfo(), const WeightsInfo &weights_info = WeightsInfo()…
107 FullyConnectedLayerInfo fc_info, WeightsInfo weights_info);
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLDeconvolutionLayer.h73 …LTensor *output, const PadStrideInfo &deconv_info, const WeightsInfo &weights_info = WeightsInfo()…
86 const WeightsInfo &weights_info = WeightsInfo());
99 const WeightsInfo &weights_info = WeightsInfo());
102 const WeightsInfo &weights_info);
DCLConvolutionLayer.h124 …ICLTensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info = WeightsInfo(),
149 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
174 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
196 …const WeightsInfo &weights_info, const ActivationLayerInfo &act_info, const GPUTarget gpu_target, …
DCLDirectDeconvolutionLayer.h114 …ias, ICLTensor *output, const PadStrideInfo &info, const WeightsInfo &weights_info = WeightsInfo()…
129 const WeightsInfo &weights_info = WeightsInfo());
144 const WeightsInfo &weights_info = WeightsInfo());
DCLGEMMConvolutionLayer.h99 …ICLTensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info = WeightsInfo(),
123 const WeightsInfo &weights_info = WeightsInfo(),
148 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
/external/ComputeLibrary/arm_compute/runtime/
DFunctionDescriptors.h66 const WeightsInfo &weights_info = WeightsInfo())
67 …fast_math(enable_fast_math), num_groups(num_groups), post_ops(post_ops), weights_info(weights_info) in conv_info()
77 WeightsInfo weights_info{};
/external/ComputeLibrary/src/gpu/cl/operators/
DClConv2d.cpp82 const WeightsInfo &weights_info) in configure() argument
85 …:validate(src, weights, ((biases != nullptr) ? biases : nullptr), dst, conv2d_info, weights_info)); in configure()
86 ARM_COMPUTE_LOG_PARAMS(src, weights, biases, dst, conv2d_info, weights_info); in configure()
88 …switch(ClConv2d::get_convolution_method(src, weights, dst, conv2d_info, weights_info, CLScheduler:… in configure()
120 f->configure(compile_context, src, weights, biases, dst, conv2d_info, weights_info); in configure()
132 const WeightsInfo &weights_info) in validate() argument
139 … switch(ClConv2d::get_convolution_method(src, weights, dst, conv2d_info, weights_info, gpu_target)) in validate()
168 …PUTE_RETURN_ON_ERROR(ClGemmConv2d::validate(src, weights, biases, dst, conv2d_info, weights_info)); in validate()
180 … const WeightsInfo &weights_info, const GPUTarget gpu_target) in get_convolution_method() argument
185 ARM_COMPUTE_UNUSED(weights_info); in get_convolution_method()
DClConv2d.h116 const WeightsInfo &weights_info = WeightsInfo());
124 const WeightsInfo &weights_info = WeightsInfo());
141 … const WeightsInfo &weights_info, const GPUTarget gpu_target);
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEConvolutionLayer.h121 …, ITensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info = WeightsInfo(),
146 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
168 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
DNEGEMMConvolutionLayer.h97 …, ITensor *output, const PadStrideInfo &conv_info, const WeightsInfo &weights_info = WeightsInfo(),
122 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
182 …const WeightsInfo &weights_info = WeightsInfo(), const Size2D &dilation = Size2D(1U, 1U), const Ac…
DNEFullyConnectedLayer.h129 …ctedLayerInfo fc_info = FullyConnectedLayerInfo(), const WeightsInfo &weights_info = WeightsInfo()…
137 …ctedLayerInfo fc_info = FullyConnectedLayerInfo(), const WeightsInfo &weights_info = WeightsInfo()…
152 …onst ITensorInfo *output, const FullyConnectedLayerInfo &fc_info, const WeightsInfo &weights_info);
/external/ComputeLibrary/tests/validation/fixtures/
DDeconvolutionLayerFixture.h259 TensorInfo weights_info(weights_shape, 1, data_type); in setup()
260 …Shape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights_info); in setup()
280 TensorInfo weights_info(weights_shape, 1, data_type); in setup()
281 …Shape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights_info); in setup()
301 TensorInfo weights_info(weights_shape, 1, data_type, input_quantization_info); in setup()
302 …Shape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights_info); in setup()
324 … TensorInfo weights_info(weights_shape, 1, weights_data_type, input_quantization_info); in setup()
325 …Shape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights_info); in setup()

123