Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser.cc1012 optional<std::vector<int64>> lhs_contracting_dims; in ParseInstruction() local
1014 /*required=*/false, AttrTy::kBracedInt64List, &lhs_contracting_dims}; in ParseInstruction()
1031 if (lhs_contracting_dims) { in ParseInstruction()
1033 lhs_contracting_dims->begin(), lhs_contracting_dims->end()}; in ParseInstruction()
Dhlo_parser_test.cc860 …ROOT dot = f32[2,3]{1,0} dot(a, b), lhs_batch_dims={0}, lhs_contracting_dims={1}, rhs_contracting_…