Searched refs:AssignStreams (Results 1 – 5 of 5) sorted by relevance
61 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module); in TEST_F()82 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module); in TEST_F()126 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module); in TEST_F()
41 std::unique_ptr<StreamAssignment> AssignStreams(const HloModule& module);
85 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()145 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()215 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()294 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()
127 std::unique_ptr<StreamAssignment> AssignStreams(const HloModule& module) { in AssignStreams() function
522 AssignStreams(*hlo_module); in RunHloPassesAndBufferAssignement()565 AssignStreams(*hlo_module); in CompileModuleToLlvmIrImpl()