Searched refs:_reshape_rhs_kernel (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemm.cpp | 190 _reshape_rhs_kernel(std::make_unique<ClGemmReshapeRhsMatrixKernel>()), in ClGemm()
271 _reshape_rhs_kernel->configure(compile_context, b, &_tmp_b, rhs_info); in configure_reshaped()
314 _reshape_rhs_kernel->configure(compile_context, b, &_tmp_b, rhs_info); in configure_reshaped_only_rhs()
365 _reshape_rhs_kernel->configure(compile_context, b, &_tmp_b, rhs_info); in configure_reshaped_only_rhs_mmul()
698 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, false); in run()
717 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, false); in run()
745 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, false); in run()
790 CLScheduler::get().enqueue_op(*_reshape_rhs_kernel, reshape_rhs_pack, true); in prepare()
|
| D | ClGemm.h | 127 std::unique_ptr<kernels::ClGemmReshapeRhsMatrixKernel> _reshape_rhs_kernel; variable
|