Searched refs:rhs_contracting_dims (Results 1 – 2 of 2) sorted by relevance
1565 optional<std::vector<int64>> rhs_contracting_dims; in ParseInstructionRhs() local1567 /*required=*/false, AttrTy::kBracedInt64List, &rhs_contracting_dims}; in ParseInstructionRhs()1588 if (rhs_contracting_dims) { in ParseInstructionRhs()1590 rhs_contracting_dims->begin(), rhs_contracting_dims->end()}; in ParseInstructionRhs()
1321 …t = f32[2,3]{1,0} dot(a, b), lhs_batch_dims={0}, lhs_contracting_dims={1}, rhs_contracting_dims={0} in CreateTestCases()