Searched defs:mat2_arg (Results 1 – 1 of 1) sorted by relevance
894 Tensor mm(const Tensor& mat1_arg, const Tensor& mat2_arg) { in mm()906 Tensor bmm(const Tensor& mat1_arg, const Tensor& mat2_arg) { in bmm()