Home
last modified time | relevance | path

Searched refs:kMainExecutionThread (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_computation.h356 HloInstruction::kMainExecutionThread);
652 return execution_thread_ == HloInstruction::kMainExecutionThread; in IsMainThread()
738 std::string execution_thread_ = HloInstruction::kMainExecutionThread;
H A Dhlo_module_test.cc631 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()
H A Dhlo_schedule_test.cc393 /*algorithm=*/{}, {HloInstruction::kMainExecutionThread})); in TEST_F()
412 TF_ASSERT_OK(schedule.Update({HloInstruction::kMainExecutionThread})); in TEST_F()
H A Dhlo_schedule.cc247 execution_threads_, it->first, HloInstruction::kMainExecutionThread); in Update()
277 HloInstruction::kMainExecutionThread)]; in num_sequences_by_execution_thread()
H A Dhlo_instruction.h509 inline static constexpr char kMainExecutionThread[] = "main"; variable
633 absl::string_view async_execution_thread = kMainExecutionThread);
638 absl::string_view async_execution_thread = kMainExecutionThread);
643 absl::string_view async_execution_thread = kMainExecutionThread);
H A Dhlo_instructions.h218 absl::string_view async_execution_thread = kMainExecutionThread);
223 absl::string_view async_execution_thread = kMainExecutionThread);
271 std::string async_execution_thread_ = kMainExecutionThread;
H A Dhlo_instruction.cc202 ? kMainExecutionThread in CreateFromProto()
218 ? kMainExecutionThread in CreateFromProto()
233 ? kMainExecutionThread in CreateFromProto()
H A Dhlo_instructions.cc311 if (async_execution_thread_ != kMainExecutionThread) { in ExtraAttributesToStringImpl()
366 HloInstruction::kMainExecutionThread in ToProto()
H A Dhlo_parser.cc1048 optional<std::string> execution_thread = HloInstruction::kMainExecutionThread; in ParseComputation()
1622 HloInstruction::kMainExecutionThread; in CreateInstruction()