Searched refs:matmul_result_buffer (Results 1 – 1 of 1) sorted by relevance
357 Tensor matmul_result_buffer; in SparseDenseMatMulWithTransposedLHS() local361 &matmul_result_buffer)); in SparseDenseMatMulWithTransposedLHS()363 set_zero(device, matmul_result_buffer.flat<T>()); in SparseDenseMatMulWithTransposedLHS()397 matmul_result_buffer.flat<T>().data() + in SparseDenseMatMulWithTransposedLHS()417 output->flat<T>().device(device) = matmul_result_buffer.matrix<T>().reduce( in SparseDenseMatMulWithTransposedLHS()