| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuFullyConnected.cpp | 266 const ITensorInfo *weights_to_use = weights; in configure() local
285 weights_to_use = &_reshaped_weights; in configure()
294 _convert_weights->configure(weights_to_use, in configure()
299 weights_to_use = &_converted_weights; in configure()
307 configure_conv_fc(src, weights_to_use, biases, dst, fc_info.activation_info); in configure()
312 configure_fc_fc(src, weights_to_use, biases, dst, fc_info.activation_info); in configure()
318 _trans_weights = *weights_to_use; in configure()
394 const ITensorInfo *weights_to_use = weights; in validate() local
425 weights_to_use = &reshaped_weights; in validate()
431 ARM_COMPUTE_RETURN_ON_ERROR(CpuConvertFullyConnectedWeights::validate(weights_to_use, in validate()
[all …]
|
| D | CpuDepthwiseConv2d.cpp | 292 const ITensorInfo *weights_to_use = weights; in configure() local
310 weights_to_use = weights_perm.get(); in configure()
316 _depthwise_conv_kernel->configure(input_to_use, weights_to_use, biases, output_to_use, info); in configure()
|
| D | CpuGemmConv2d.cpp | 454 const ITensorInfo *weights_to_use = weights; in validate() local
503 weights_to_use = &weights_reshaped_info; in validate()
537 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemm_input_to_use, weights_to_use, biases, gemm_output_to_… in validate()
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClFullyConnected.cpp | 353 const ITensorInfo *weights_to_use = weights; in validate() local
385 weights_to_use = &reshaped_weights; in validate()
391 ARM_COMPUTE_RETURN_ON_ERROR(ClConvertFullyConnectedWeights::validate(weights_to_use, in validate()
395 weights_to_use = &converted_weights; in validate()
401 …ARM_COMPUTE_RETURN_ERROR_ON((weights_to_use->dimension(1) != (src->dimension(0) * src->dimension(1… in validate()
410 ARM_COMPUTE_RETURN_ERROR_ON(src->dimension(0) != weights_to_use->dimension(1)); in validate()
414 ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(*src_to_use, *weights_to_use, biases, *dst, fc_info)); in validate()
|
| D | ClGemmConv2d.cpp | 382 const ITensorInfo *weights_to_use = weights; in validate() local
445 weights_to_use = &weights_reshaped_info; in validate()
523 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemm_input_to_use, weights_to_use, biases_to_use, gemm_out… in validate()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLGEMMDeconvolutionLayer.cpp | 241 const ICLTensor *weights_to_use = weights; in configure() local
255 weights_to_use = &_permuted_weights; in configure()
259 _reshaped_weights.allocator()->init(TensorInfo(TensorShape(weights_to_use->info()->dimension(0), in configure()
260 …weights_to_use->info()->dimension(1) * weights_to_use->info()->dimension(2) * weights_to_use->info… in configure()
264 _reshape_weights.configure(compile_context, weights_to_use, &_reshaped_weights); in configure()
|
| D | CLDepthwiseConvolutionLayer.cpp | 97 const ICLTensor *weights_to_use = weights; in configure() local
116 weights_to_use = &_permuted_weights; in configure()
136 …nelInfo dwc_native_compute_info = t->configure(input_to_use->info(), weights_to_use->info(), conv_… in configure()
141 …_dwc_native_kernel->configure(compile_context, input_to_use, weights_to_use, biases, output_to_use, in configure()
|
| D | CLFFTConvolutionLayer.cpp | 137 const ICLTensor *weights_to_use = weights; in configure() local
161 weights_to_use = &_permuted_weights; in configure()
165 …_flipped_weights.allocator()->init(weights_to_use->info()->clone()->set_is_resizable(true).reset_p… in configure()
167 _flip_weights_func.configure(compile_context, weights_to_use, &_flipped_weights, &_flip_axis); in configure()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEFFTConvolutionLayer.cpp | 128 const ITensor *weights_to_use = weights; in configure() local
152 weights_to_use = &_permuted_weights; in configure()
156 …_flipped_weights.allocator()->init(weights_to_use->info()->clone()->set_is_resizable(true).reset_p… in configure()
158 _flip_weights_func.configure(weights_to_use, &_flipped_weights, &_flip_axis); in configure()
|
| D | NEDepthwiseConvolutionLayer.cpp | 239 const ITensor *weights_to_use = weights; in configure() local
252 weights_to_use = &_impl->permuted_weights; in configure()
259 …depthwise_conv_kernel->configure(input_to_use->info(), weights_to_use->info(), biases == nullptr ?… in configure()
|