Home
last modified time | relevance | path

Searched refs:async_execution_thread (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.cc201 proto.async_execution_thread().empty() in CreateFromProto()
203 : proto.async_execution_thread()); in CreateFromProto()
217 proto.async_execution_thread().empty() in CreateFromProto()
219 : proto.async_execution_thread()); in CreateFromProto()
232 proto.async_execution_thread().empty() in CreateFromProto()
234 : proto.async_execution_thread()); in CreateFromProto()
1190 absl::string_view async_execution_thread) { in CreateAsyncStart() argument
1193 async_group_id, async_execution_thread); in CreateAsyncStart()
1199 absl::string_view async_execution_thread) { in CreateAsyncUpdate() argument
1202 async_group_id, async_execution_thread); in CreateAsyncUpdate()
[all …]
H A Dhlo_instruction.h633 absl::string_view async_execution_thread = kMainExecutionThread);
638 absl::string_view async_execution_thread = kMainExecutionThread);
643 absl::string_view async_execution_thread = kMainExecutionThread);
2139 absl::string_view async_execution_thread() const;
2142 void set_async_execution_thread(absl::string_view async_execution_thread);
2147 absl::string_view async_execution_thread,
H A Dhlo_instructions.cc247 absl::string_view async_execution_thread) in HloAsyncInstruction() argument
250 async_execution_thread_(async_execution_thread) { in HloAsyncInstruction()
259 set_async_execution_thread(async_execution_thread); in HloAsyncInstruction()
265 absl::string_view async_execution_thread) in HloAsyncInstruction() argument
268 async_execution_thread_(async_execution_thread) { in HloAsyncInstruction()
274 set_async_execution_thread(async_execution_thread); in HloAsyncInstruction()
356 absl::string_view async_execution_thread) { in set_async_execution_thread() argument
357 async_execution_thread_ = std::string(async_execution_thread); in set_async_execution_thread()
358 SetThreadName(async_wrapped_computation(), async_execution_thread, in set_async_execution_thread()
H A Dhlo_constant_folding.cc143 instruction->async_execution_thread() != in Run()
H A Dhlo_parser_test.cc1290 …2[10]{0}), f32[20]{0}, s32[]) custom-call-start(f32[10]{0} %p0), async_execution_thread="parallel_… in CreateTestCases()
1291 …tom-call-update(((f32[10]{0}), f32[20]{0}, s32[]) %async-start), async_execution_thread="parallel_… in CreateTestCases()
1292 …stom-call-done(((f32[10]{0}), f32[20]{0}, s32[]) %async-update), async_execution_thread="parallel_… in CreateTestCases()
1304 …2[10]{0}), f32[20]{0}, s32[]) custom-call-start(f32[10]{0} %p0), async_execution_thread="parallel_… in CreateTestCases()
1305 …tom-call-update(((f32[10]{0}), f32[20]{0}, s32[]) %async-start), async_execution_thread="parallel_… in CreateTestCases()
1306 …stom-call-done(((f32[10]{0}), f32[20]{0}, s32[]) %async-update), async_execution_thread="parallel_… in CreateTestCases()
H A Dhlo_verifier.cc1419 std::optional<absl::string_view> async_execution_thread = in CheckAsyncOpComputationThreadName() local
1420 async_op->async_execution_thread(); in CheckAsyncOpComputationThreadName()
1421 if (async_execution_thread != in CheckAsyncOpComputationThreadName()
1426 async_execution_thread ? absl::StrCat(*async_execution_thread) : "none", in CheckAsyncOpComputationThreadName()
H A Dhlo_computation.cc761 absl::string_view async_execution_thread) { in CreateAsyncInstructions() argument
783 async_execution_thread)); in CreateAsyncInstructions()
786 /*async_group_id=*/std::nullopt, async_execution_thread)); in CreateAsyncInstructions()
H A Dhlo_instructions.h218 absl::string_view async_execution_thread = kMainExecutionThread);
223 absl::string_view async_execution_thread = kMainExecutionThread);
242 absl::string_view async_execution_thread() const { in async_execution_thread() function
246 void set_async_execution_thread(absl::string_view async_execution_thread);
H A Dhlo_computation.h355 absl::string_view async_execution_thread =
H A Dhlo.proto338 // Empty async_execution_thread is equivalent to main thread.
339 string async_execution_thread = 79; field
H A Dhlo_parser.cc1621 optional<std::string> async_execution_thread = in CreateInstruction() local
1624 &async_execution_thread}; in CreateInstruction()
1686 *async_execution_thread)); in CreateInstruction()
1691 *async_execution_thread)); in CreateInstruction()
1695 *async_execution_thread)); in CreateInstruction()
H A Dhlo_instruction_test.cc798 EXPECT_EQ(async_start->async_execution_thread(), "parallel_thread"); in TEST_F()
799 EXPECT_EQ(async_done->async_execution_thread(), "parallel_thread"); in TEST_F()