Home
last modified time | relevance | path

Searched defs:async_computation (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.cc1189 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncStart()
1198 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncUpdate()
1207 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncDone()
H A Dhlo_computation.cc773 HloComputation* async_computation = in CreateAsyncInstructions() local
H A Dhlo_instructions.cc246 HloComputation* async_computation, std::optional<int64_t> async_group_id, in HloAsyncInstruction()
264 HloComputation* async_computation, std::optional<int64_t> async_group_id, in HloAsyncInstruction()
H A Dhlo_instruction_test.cc1717 std::unique_ptr<HloComputation> async_computation = async_builder.Build(); in TEST_F() local
H A Dhlo_parser.cc1610 std::optional<HloComputation*> async_computation; in CreateInstruction() local