Searched refs:_weights_reshaped (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemmConv2d.cpp | 56 …lptr), _col2im_kernel(nullptr), _activation_kernel(nullptr), _im2col_output(), _weights_reshaped(), in ClGemmConv2d()
224 …_weights_reshape_kernel->configure(compile_context, weights, biases, &_weights_reshaped, conv2d_in… in configure()
228 …_weights_reshape_kernel->configure(compile_context, weights, nullptr, &_weights_reshaped, conv2d_i… in configure()
321 …configure_mm(compile_context, gemm_input_to_use, &_weights_reshaped, biases_to_use, gemm_output_to… in configure()
345 …ryInfo(offset_int_vec(WeightsReshaped), MemoryLifetime::Persistent, _weights_reshaped.total_size()… in configure()
554 …CLAuxTensorHandler weights_reshaped(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors, f… in run()
621 CLAuxTensorHandler weights_reshaped(_weights_reshaped, *weights_reshaped_p); in prepare()
|
| D | ClGemmConv2d.h | 172 TensorInfo _weights_reshaped; variable
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuGemmConv2d.cpp | 95 …_gemmlowp(), _col2im_kernel(), _reshape_kernel(), _im2col_output(), _weights_reshaped(), _gemm_out… in CpuGemmConv2d()
318 _weights_reshape_kernel->configure(weights, nullptr, &_weights_reshaped); in configure()
319 _weights_reshaped.set_quantization_info(weights->quantization_info()); in configure()
364 …configure_mm(gemm_input_to_use, &_weights_reshaped, biases, gemm_output_to_use, act_info, enable_f… in configure()
387 …m_trans_wei ? MemoryLifetime::Prepare : MemoryLifetime::Persistent, _weights_reshaped.total_size()… in configure()
559 …CpuAuxTensorHandler reshaped_wei(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors, fals… in run()
658 … CpuAuxTensorHandler weights_reshaped(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors); in prepare()
|
| D | CpuGemmConv2d.h | 227 TensorInfo _weights_reshaped; variable
|