Home
last modified time | relevance | path

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

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