Home
last modified time | relevance | path

Searched refs:lhs_batch_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc1568 optional<std::vector<int64>> lhs_batch_dims; in ParseInstructionRhs() local
1570 &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()
Dhlo_parser_test.cc1321 …ROOT dot = f32[2,3]{1,0} dot(a, b), lhs_batch_dims={0}, lhs_contracting_dims={1}, rhs_contracting_… in CreateTestCases()