Searched refs:operand_layout_constraints (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.cc | 1490 optional<std::vector<Shape>> operand_layout_constraints; in ParseInstructionRhs() local
1502 /*required=*/false, AttrTy::kShapeList, &operand_layout_constraints}; in ParseInstructionRhs()
1506 if (operand_layout_constraints.has_value()) { in ParseInstructionRhs()
1511 if (operands.size() != operand_layout_constraints->size()) { in ParseInstructionRhs()
1515 operand_layout_constraints->size(), " given")); in ParseInstructionRhs()
1519 (*operand_layout_constraints)[i]; in ParseInstructionRhs()
1540 shape, operands, *custom_call_target, *operand_layout_constraints, in ParseInstructionRhs()
|
| D | hlo_parser_test.cc | 906 …3]{0,1,2} %p0, f32[123,4]{0,1} %p1), custom_call_target="baz", operand_layout_constraints={f32[42,… in CreateTestCases()
917 …tom-call = f32[1,2,3]{0,2,1} custom-call(), custom_call_target="baz", operand_layout_constraints={} in CreateTestCases()
930 …]{0,1,2}) %p0, f32[123,4]{0,1} %p1), custom_call_target="baz", operand_layout_constraints={(f32[2,… in CreateTestCases()
|