Searched refs:gemm_reshaped_pack (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemmLowpMatrixMultiplyCore.cpp | 718 ITensorPack gemm_reshaped_pack; in run() local
721 gemm_reshaped_pack = ITensorPack({ { TensorType::ACL_SRC_0, matrix_a }, in run()
728 gemm_reshaped_pack = ITensorPack( in run()
742 CLScheduler::get().enqueue_op(*_mm_reshaped_only_rhs_kernel, gemm_reshaped_pack, false); in run()
746 … CLScheduler::get().enqueue_op(*_mm_reshaped_only_rhs_mmul_kernel, gemm_reshaped_pack, false); in run()
|
| D | ClGemm.cpp | 701 ITensorPack gemm_reshaped_pack(tensors); in run() local
702 gemm_reshaped_pack.add_const_tensor(ACL_SRC_0, lhs_reshaped.get()); in run()
703 gemm_reshaped_pack.add_const_tensor(ACL_SRC_1, rhs_reshaped.get()); in run()
707 CLScheduler::get().enqueue_op(*_mm_reshaped_kernel, gemm_reshaped_pack, true); in run()
|