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.cc659 absl::flat_hash_map<int64_t, HloInstruction*> instruction_map; in CreateFromProto() local
666 instruction_proto, instruction_map, computation_map, in CreateFromProto()
671 TF_RET_CHECK(!ContainsKey(instruction_map, instruction_proto.id())); in CreateFromProto()
672 instruction_map[instruction_proto.id()] = instruction.get(); in CreateFromProto()
678 TF_RET_CHECK(ContainsKey(instruction_map, proto.root_id())); in CreateFromProto()
679 HloInstruction* root = instruction_map.at(proto.root_id()); in CreateFromProto()
Dhlo_instruction.cc86 const absl::flat_hash_map<int64_t, HloInstruction*>& instruction_map, in CreateFromProto() argument
122 const auto operands = [&instruction_map, &proto](int index) { in CreateFromProto()
123 return instruction_map.at(proto.operand_ids(index)); in CreateFromProto()
125 const auto all_operands = [&instruction_map, &proto]() { in CreateFromProto()
128 result.begin(), [&instruction_map](int64_t operand_id) { in CreateFromProto()
129 return instruction_map.at(operand_id); in CreateFromProto()
148 [&](int64_t id) { return instruction_map.contains(id); })) in CreateFromProto()
948 instruction->AppendOperand(instruction_map.at(operand_id)); in CreateFromProto()
974 TF_RET_CHECK(ContainsKey(instruction_map, predecessor_id)) in CreateFromProto()
976 TF_RETURN_IF_ERROR(instruction_map.at(predecessor_id) in CreateFromProto()
Dhlo_instruction.h532 const absl::flat_hash_map<int64_t, HloInstruction*>& instruction_map,