Home
last modified time | relevance | path

Searched refs:async_wrapped_computation (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dasync_op_canonicalizer_test.cc71 EXPECT_EQ(async_start->async_wrapped_computation(), in TEST_F()
72 async_update->async_wrapped_computation()); in TEST_F()
73 EXPECT_EQ(async_start->async_wrapped_computation(), in TEST_F()
74 async_done->async_wrapped_computation()); in TEST_F()
134 EXPECT_EQ(async_start->async_wrapped_computation(), in TEST_F()
135 async_update->async_wrapped_computation()); in TEST_F()
136 EXPECT_EQ(async_start->async_wrapped_computation(), in TEST_F()
137 async_done->async_wrapped_computation()); in TEST_F()
H A Dflatten_call_graph_test.cc294 ->async_wrapped_computation(), in TEST_F()
296 ->async_wrapped_computation()); in TEST_F()
298 ->async_wrapped_computation(), in TEST_F()
300 ->async_wrapped_computation()); in TEST_F()
302 ->async_wrapped_computation(), in TEST_F()
304 ->async_wrapped_computation()); in TEST_F()
H A Dasync_op_canonicalizer.cc106 async_group.instructions[0]->async_wrapped_computation(); in Run()
109 if (instruction->async_wrapped_computation() != computation) { in Run()
110 instruction->async_wrapped_computation()->RemoveAsyncInstruction( in Run()
H A Dhlo_pass_pipeline_test.cc204 main_root->async_wrapped_computation()->root_instruction(); in TEST_F()
240 main_root->async_wrapped_computation()->root_instruction(); in TEST_F()
H A Dhlo_verifier.cc1364 if (*async_op->async_wrapped_computation() != in CheckAsyncOpOperand()
1365 *operand->async_wrapped_computation()) { in CheckAsyncOpOperand()
1373 async_op->async_wrapped_computation()->execution_thread(), in CheckAsyncOpOperand()
1374 operand->async_wrapped_computation()->execution_thread()); in CheckAsyncOpOperand()
1397 async_op->async_wrapped_computation()->ComputeProgramShape(); in CheckAsyncOpComputationShapes()
1422 async_op->async_wrapped_computation()->execution_thread()) { in CheckAsyncOpComputationThreadName()
1427 async_op->async_wrapped_computation()->execution_thread()); in CheckAsyncOpComputationThreadName()
1430 async_op->async_wrapped_computation(), in CheckAsyncOpComputationThreadName()
H A Dhlo_dataflow_analysis.cc631 async_start->async_wrapped_computation()->root_instruction(); in UpdateAsyncStartValueSet()
657 async_update->async_wrapped_computation()->root_instruction(); in UpdateAsyncUpdateValueSet()
692 async_done->async_wrapped_computation()->root_instruction(); in UpdateAsyncDoneValueSet()
1270 user->async_wrapped_computation()->num_parameters(); in Propagate()
1273 user->async_wrapped_computation()->parameter_instruction( in Propagate()
H A Dhlo_instructions.cc328 eq_computations(async_wrapped_computation(), in IdenticalSlowPath()
329 other.async_wrapped_computation()); in IdenticalSlowPath()
339 context->FindComputation(async_wrapped_computation()); in CloneWithNewOperandsImpl()
343 async_wrapped_computation()->Clone("clone", context)); in CloneWithNewOperandsImpl()
358 SetThreadName(async_wrapped_computation(), async_execution_thread, in set_async_execution_thread()
H A Dhlo_module_test.cc628 auto* parallel_thread_computation = async_done->async_wrapped_computation(); in TEST_F()
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_computation.cc114 CHECK(async_instr->async_wrapped_computation() == this); in ~HloComputation()
H A Dhlo_instruction.h2124 HloComputation* async_wrapped_computation() const;
H A Dhlo_instruction.cc3258 PrintNameInternal(async_wrapped_computation()->name(), options))); in ExtraAttributesToString()
4849 HloComputation* HloInstruction::async_wrapped_computation() const { in async_wrapped_computation() function in xla::HloInstruction
H A Dhlo_evaluator.cc3047 embedded_evaluator.Evaluate(*async_start->async_wrapped_computation(), in HandleAsyncStart()