Searched defs:instruction_map (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_computation.cc | 413 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 D | hlo_instruction.cc | 54 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