Searched refs:kMainExecutionThread (Results 1 – 9 of 9) sorted by relevance
356 HloInstruction::kMainExecutionThread);652 return execution_thread_ == HloInstruction::kMainExecutionThread; in IsMainThread()738 std::string execution_thread_ = HloInstruction::kMainExecutionThread;
631 module->MakeComputationPostOrder({HloInstruction::kMainExecutionThread}), in TEST_F()647 module->computations({HloInstruction::kMainExecutionThread})) { in TEST_F()649 EXPECT_EQ(comp->execution_thread(), HloInstruction::kMainExecutionThread); in TEST_F()
393 /*algorithm=*/{}, {HloInstruction::kMainExecutionThread})); in TEST_F()412 TF_ASSERT_OK(schedule.Update({HloInstruction::kMainExecutionThread})); in TEST_F()
247 execution_threads_, it->first, HloInstruction::kMainExecutionThread); in Update()277 HloInstruction::kMainExecutionThread)]; in num_sequences_by_execution_thread()
509 inline static constexpr char kMainExecutionThread[] = "main"; variable633 absl::string_view async_execution_thread = kMainExecutionThread);638 absl::string_view async_execution_thread = kMainExecutionThread);643 absl::string_view async_execution_thread = kMainExecutionThread);
218 absl::string_view async_execution_thread = kMainExecutionThread);223 absl::string_view async_execution_thread = kMainExecutionThread);271 std::string async_execution_thread_ = kMainExecutionThread;
202 ? kMainExecutionThread in CreateFromProto()218 ? kMainExecutionThread in CreateFromProto()233 ? kMainExecutionThread in CreateFromProto()
311 if (async_execution_thread_ != kMainExecutionThread) { in ExtraAttributesToStringImpl()366 HloInstruction::kMainExecutionThread in ToProto()
1048 optional<std::string> execution_thread = HloInstruction::kMainExecutionThread; in ParseComputation()1622 HloInstruction::kMainExecutionThread; in CreateInstruction()