Searched refs:lhs_batch_dims (Results 1 – 2 of 2) sorted by relevance
1568 optional<std::vector<int64>> lhs_batch_dims; in ParseInstructionRhs() local1570 &lhs_batch_dims}; in ParseInstructionRhs()1592 if (lhs_batch_dims) { in ParseInstructionRhs()1593 *dnum.mutable_lhs_batch_dimensions() = {lhs_batch_dims->begin(), in ParseInstructionRhs()1594 lhs_batch_dims->end()}; in ParseInstructionRhs()
1321 …ROOT dot = f32[2,3]{1,0} dot(a, b), lhs_batch_dims={0}, lhs_contracting_dims={1}, rhs_contracting_… in CreateTestCases()