Searched defs:num_lhs_rows (Results 1 – 1 of 1) sorted by relevance
180 int64 num_lhs_rows = dense_shape(rank - 2); in Compute() local288 OpKernelContext* ctx, const int64 batch_size, const int64 num_lhs_rows, in SparseDenseMatMulWithoutTransposedLHS()337 const int64 num_lhs_rows, in SparseDenseMatMulWithTransposedLHS()