Home
last modified time | relevance | path

Searched refs:instruction_map (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_computation.cc538 absl::flat_hash_map<int64, HloInstruction*> instruction_map; in CreateFromProto() local
545 HloInstruction::CreateFromProto(instruction_proto, instruction_map, in CreateFromProto()
550 TF_RET_CHECK(!ContainsKey(instruction_map, instruction_proto.id())); in CreateFromProto()
551 instruction_map[instruction_proto.id()] = instruction.get(); in CreateFromProto()
557 TF_RET_CHECK(ContainsKey(instruction_map, proto.root_id())); in CreateFromProto()
558 HloInstruction* root = instruction_map.at(proto.root_id()); in CreateFromProto()
Dhlo_instruction.cc64 const absl::flat_hash_map<int64, HloInstruction*>& instruction_map, in CreateFromProto() argument
99 const auto operands = [&instruction_map, &proto](int index) { in CreateFromProto()
100 return instruction_map.at(proto.operand_ids(index)); in CreateFromProto()
102 const auto all_operands = [&instruction_map, &proto]() { in CreateFromProto()
105 result.begin(), [&instruction_map](int64 operand_id) { in CreateFromProto()
106 return instruction_map.at(operand_id); in CreateFromProto()
125 [&](int64 id) { return instruction_map.contains(id); })) in CreateFromProto()
595 instruction->AppendOperand(instruction_map.at(operand_id)); in CreateFromProto()
611 TF_RET_CHECK(ContainsKey(instruction_map, predecessor_id)) in CreateFromProto()
613 TF_RETURN_IF_ERROR(instruction_map.at(predecessor_id) in CreateFromProto()
Dhlo_instruction.h364 const absl::flat_hash_map<int64, HloInstruction*>& instruction_map,