Home
last modified time | relevance | path

Searched refs:execution_thread (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_verifier.cc131 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 Dhlo_computation.cc630 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 Dhlo_computation.h645 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 Dhlo_schedule.cc104 std::string(computation->execution_thread()); in set_sequence()
114 std::string(computation->execution_thread()); in GetOrCreateSequence()
H A Dhlo_module_test.cc649 EXPECT_EQ(comp->execution_thread(), HloInstruction::kMainExecutionThread); in TEST_F()
656 EXPECT_EQ(comp->execution_thread(), kParallelThreadName); in TEST_F()
H A Dhlo_constant_folding.cc144 instruction->parent()->execution_thread()) { in Run()
H A Dhlo_instructions.cc89 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 Dhlo_module.h231 return execution_threads.contains(computation->execution_thread()); in computations()
H A Dhlo_rematerialization.cc1878 computation->execution_thread())) { in CalledComputationsMemoryUsage()
2127 node.computation()->execution_thread())) { in Run()
H A Dinstruction_fusion.cc721 computation->execution_thread(), in AddFusionInstruction()
H A Dhlo_module.cc792 return execution_threads.find(computation->execution_thread()) != in MakeComputationPostOrder()
H A Dhlo_instruction_test.cc802 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 Dhlo.proto369 string execution_thread = 8; field
H A Dhlo_parser.cc1048 optional<std::string> execution_thread = HloInstruction::kMainExecutionThread; in ParseComputation() local
1050 &execution_thread}; in ParseComputation()
1054 computation->SetExecutionThread(*execution_thread); in ParseComputation()
H A Dhlo_instructions.h1193 absl::string_view execution_thread,
H A Dhlo_parser_test.cc1307 }, execution_thread="main_thread" in CreateTestCases()