Searched defs:instruction_map (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_computation.cc413 tensorflow::gtl::FlatMap<string, HloInstruction*> instruction_map; local
418 module, instruction_proto, instruction_map,
423 TF_RET_CHECK(!ContainsKey(instruction_map, instruction->name()));
424 instruction_map[instruction->name()] = instruction.get();
429 TF_RET_CHECK(ContainsKey(instruction_map, proto.root_name()));
430 HloInstruction* root = instruction_map.at(proto.root_name());
H A Dhlo_instruction.cc54 const tensorflow::gtl::FlatMap<string, HloInstruction*>& instruction_map,
64 TF_RET_CHECK(ContainsKey(instruction_map, operand_name))
66 instruction->AppendOperand(instruction_map.at(operand_name));
69 TF_RET_CHECK(ContainsKey(instruction_map, predecessor_name))
71 TF_RETURN_IF_ERROR(instruction_map.at(predecessor_name)
52 CreateFromProto( HloModule* module, const HloInstructionProto& proto, const tensorflow::gtl::FlatMap<string, HloInstruction*>& instruction_map, const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map, const std::function<void(std::unique_ptr<HloComputation>)>& add_fused_computation) argument

Completed in 309 milliseconds