Home
last modified time | relevance | path

Searched refs:output_to_use (Results 1 – 6 of 6) sorted by relevance

/external/ComputeLibrary/src/runtime/NEON/functions/
DNEFFTConvolutionLayer.cpp129 ITensor *output_to_use = _has_bias ? &_bias_output : output; in configure() local
216 output_to_use = &_permuted_output; in configure()
219 …_extract_output_func.configure(&_reshaped_output, output_to_use, Coordinates(start_left, start_top… in configure()
226 output_to_use = output; in configure()
229 output_to_use = &_permuted_output; in configure()
232 auto_init_if_empty(*output_to_use->info(), *_bias_output.info()); in configure()
233 … _bias_add_func.configure(&_bias_output, &_permuted_bias, output_to_use, ConvertPolicy::WRAP); in configure()
DNEDepthwiseConvolutionLayer.cpp240 ITensor *output_to_use = output; in configure() local
255 output_to_use = &_impl->permuted_output; in configure()
259 …weights_to_use->info(), biases == nullptr ? nullptr : biases->info(), output_to_use->info(), info); in configure()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLFFTConvolutionLayer.cpp138 ICLTensor *output_to_use = _has_bias ? &_bias_output : output; in configure() local
225 output_to_use = &_permuted_output; in configure()
228 …_extract_output_func.configure(compile_context, &_reshaped_output, output_to_use, Coordinates(star… in configure()
234 output_to_use = output; in configure()
237 output_to_use = &_permuted_output; in configure()
240 auto_init_if_empty(*output_to_use->info(), *_bias_output.info()); in configure()
241 …_bias_add_func.configure(compile_context, &_bias_output, &_permuted_bias, output_to_use, ConvertPo… in configure()
DCLDepthwiseConvolutionLayer.cpp98 ICLTensor *output_to_use = output; in configure() local
117 output_to_use = &_permuted_output; in configure()
141 …_dwc_native_kernel->configure(compile_context, input_to_use, weights_to_use, biases, output_to_use, in configure()
/external/ComputeLibrary/src/cpu/operators/
DCpuGemmLowpMatrixMultiplyCore.cpp185 auto output_to_use = (_fuse_output_stage ? &_mm_result_s32 : dst); in configure() local
186 _asm_glue->configure(a_to_use, b, nullptr, output_to_use, asm_info); in configure()
542 auto output_to_use = (_fuse_output_stage ? mm_result_s32.get() : dst); in run() local
554 asm_glue_tensors.add_tensor(TensorType::ACL_DST, output_to_use); in run()
DCpuDepthwiseConv2d.cpp293 ITensorInfo *output_to_use = dst; in configure() local
312 output_to_use = output_perm.get(); in configure()
316 _depthwise_conv_kernel->configure(input_to_use, weights_to_use, biases, output_to_use, info); in configure()