Searched refs:RhsReshape (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemm.cpp | 278 …_aux_mem[RhsReshape] = MemoryInfo(offset_int_vec(RhsReshape), _reshape_b_only_on_first_run ? Memor… in configure_reshaped()
325 …_aux_mem[RhsReshape] = MemoryInfo(offset_int_vec(RhsReshape), _reshape_b_only_on_first_run ? Memor… in configure_reshaped_only_rhs()
372 …_aux_mem[RhsReshape] = MemoryInfo(offset_int_vec(RhsReshape), _reshape_b_only_on_first_run ? Memor… in configure_reshaped_only_rhs_mmul()
675 CLAuxTensorHandler rhs_reshaped(offset_int_vec(RhsReshape), _tmp_b, tensors, true); in run()
779 …= utils::cast::polymorphic_downcast<ICLTensor *>(constants.get_tensor(offset_int_vec(RhsReshape))); in prepare()
782 …if((_aux_mem[AuxTensorIdx::RhsReshape].lifetime == MemoryLifetime::Persistent) && (src1 != nullptr… in prepare()
|
| D | ClGemmLowpMatrixMultiplyCore.h | 114 RhsReshape, enumerator
|
| D | ClGemm.h | 121 RhsReshape, enumerator
|
| D | ClGemmLowpMatrixMultiplyCore.cpp | 457 …_aux_mem[RhsReshape] = MemoryInfo(offset_int_vec(RhsReshape), _reshape_b_only_on_first_run ? Memor… in configure()
667 CLAuxTensorHandler tmp_b(offset_int_vec(RhsReshape), _tmp_b, tensors, true); in run()
797 CLAuxTensorHandler tmp_b(offset_int_vec(RhsReshape), _tmp_b, tensors, true); in prepare()
|