Searched refs:instruction_list (Results 1 – 8 of 8) sorted by relevance
289 ItemList GetUsers(const InstructionList& instruction_list, in GetUsers() argument313 Item* user_item = instruction_list.GetItem(user); in GetUsers()331 const InstructionList& instruction_list);512 const InstructionList& instruction_list) in MemoryUsageTracker() argument513 : computation_(computation), instruction_list_(instruction_list) { in MemoryUsageTracker()864 const InstructionList& instruction_list, int64 memory_limit_bytes, in PickRematerializationCandidate() argument871 for (auto* item = instruction_list.first(); item != nullptr; in PickRematerializationCandidate()872 item = instruction_list.next(item)) { in PickRematerializationCandidate()937 InstructionList instruction_list(order); in ComputePeakMemory() local939 instruction_list); in ComputePeakMemory()[all …]
26 std::vector<HloInstruction*> instruction_list; in instructions_to_modify() local35 instruction_list.push_back(instruction); in instructions_to_modify()52 instruction_list.push_back(instruction); in instructions_to_modify()62 VLOG(1) << "Found " << instruction_list.size() in instructions_to_modify()65 return instruction_list; in instructions_to_modify()
16 : 'ENTRY' name param_list_to_shape instruction_list17 | name param_list_to_shape instruction_list18 | 'ENTRY' name instruction_list19 | name instruction_list24 instruction_list
51 instruction_list.h143 instruction_list.cpp
52 instruction_list.h145 instruction_list.cpp
521 "source/opt/instruction_list.cpp",522 "source/opt/instruction_list.h",
523 "source/opt/instruction_list.cpp",524 "source/opt/instruction_list.h",
113 source/opt/instruction_list.cpp \