Searched refs:execution_thread (Results 1 – 16 of 16) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_verifier.cc | 131 if (called_cmp->execution_thread() != comp->execution_thread()) { in CheckNestedComputationThreadNameEqual() 135 called_cmp->execution_thread(), comp->execution_thread()); in CheckNestedComputationThreadNameEqual() 1373 async_op->async_wrapped_computation()->execution_thread(), in CheckAsyncOpOperand() 1374 operand->async_wrapped_computation()->execution_thread()); in CheckAsyncOpOperand() 1422 async_op->async_wrapped_computation()->execution_thread()) { in CheckAsyncOpComputationThreadName() 1427 async_op->async_wrapped_computation()->execution_thread()); in CheckAsyncOpComputationThreadName() 1438 if (instruction->parent()->execution_thread() != in CheckCallableInstructionThreadName() 1439 computation->execution_thread()) { in CheckCallableInstructionThreadName() 1443 instruction->ToString(), instruction->parent()->execution_thread(), in CheckCallableInstructionThreadName() 1444 computation->execution_thread()); in CheckCallableInstructionThreadName()
|
H A D | hlo_computation.cc | 630 result.Append(StrCat(", execution_thread=\"", execution_thread(), "\"")); in ToCord() 650 : std::string(execution_thread())); in ToProto() 714 if (!proto.execution_thread().empty()) { in CreateFromProto() 715 computation->SetExecutionThread(proto.execution_thread()); in CreateFromProto() 936 return execution_thread() == other.execution_thread(); in EqualInternal() 1319 result->SetExecutionThread(execution_thread()); in CloneWithReplacements()
|
H A D | hlo_computation.h | 645 void SetExecutionThread(absl::string_view execution_thread) { in SetExecutionThread() argument 646 execution_thread_ = std::string(execution_thread); in SetExecutionThread() 649 absl::string_view execution_thread() const { return execution_thread_; } in execution_thread() function
|
H A D | hlo_schedule.cc | 104 std::string(computation->execution_thread()); in set_sequence() 114 std::string(computation->execution_thread()); in GetOrCreateSequence()
|
H A D | hlo_module_test.cc | 649 EXPECT_EQ(comp->execution_thread(), HloInstruction::kMainExecutionThread); in TEST_F() 656 EXPECT_EQ(comp->execution_thread(), kParallelThreadName); in TEST_F()
|
H A D | hlo_constant_folding.cc | 144 instruction->parent()->execution_thread()) { in Run()
|
H A D | hlo_instructions.cc | 89 absl::string_view execution_thread, in SetThreadName() argument 91 called_computation->SetExecutionThread(execution_thread); in SetThreadName() 97 instr->set_async_execution_thread(execution_thread); in SetThreadName() 103 SetThreadName(nested_called_computation, execution_thread, in SetThreadName() 1803 absl::string_view execution_thread, in RecursivelySetComputationsThreadName() argument 1806 SetThreadName(comp, execution_thread, in RecursivelySetComputationsThreadName()
|
H A D | hlo_module.h | 231 return execution_threads.contains(computation->execution_thread()); in computations()
|
H A D | hlo_rematerialization.cc | 1878 computation->execution_thread())) { in CalledComputationsMemoryUsage() 2127 node.computation()->execution_thread())) { in Run()
|
H A D | instruction_fusion.cc | 721 computation->execution_thread(), in AddFusionInstruction()
|
H A D | hlo_module.cc | 792 return execution_threads.find(computation->execution_thread()) != in MakeComputationPostOrder()
|
H A D | hlo_instruction_test.cc | 802 EXPECT_EQ(async_start->async_wrapped_computation()->execution_thread(), in TEST_F() 804 EXPECT_EQ(async_done->async_wrapped_computation()->execution_thread(), in TEST_F()
|
H A D | hlo.proto | 369 string execution_thread = 8; field
|
H A D | hlo_parser.cc | 1048 optional<std::string> execution_thread = HloInstruction::kMainExecutionThread; in ParseComputation() local 1050 &execution_thread}; in ParseComputation() 1054 computation->SetExecutionThread(*execution_thread); in ParseComputation()
|
H A D | hlo_instructions.h | 1193 absl::string_view execution_thread,
|
H A D | hlo_parser_test.cc | 1307 }, execution_thread="main_thread" in CreateTestCases()
|