Searched defs:mat_b (Results 1 – 3 of 3) sorted by relevance
145 nir_deref_instr *mat_b = vtn_get_cmat_deref(b, w[4]); in vtn_handle_cooperative_instruction() local235 nir_deref_instr *mat_b = vtn_get_cmat_deref(b, w[4]); in vtn_handle_cooperative_alu() local
416 hipblasLtMatrixLayout_t mat_a, mat_b, mat_c; in Call() local
1190 _scaled_mm_cuda(const Tensor& mat_a, const Tensor& mat_b, in _scaled_mm_cuda()