Searched refs:mat1_buffer (Results 1 – 1 of 1) sorted by relevance
105 const Tensor& mat1_buffer = nt_mat1->get_buffer(); in NestedTensor_times_Tensor_plus_Tensor_addmm() local108 self, mat1_buffer.reshape({-1, mat2.sizes()[0]}), mat2, beta, alpha) in NestedTensor_times_Tensor_plus_Tensor_addmm()111 mat1_buffer.reshape({-1, mat2.sizes()[0]}), in NestedTensor_times_Tensor_plus_Tensor_addmm()