Searched refs:async_computation (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instructions.cc | 246 HloComputation* async_computation, std::optional<int64_t> async_group_id, in HloAsyncInstruction() argument
255 AppendComputation(async_computation); in HloAsyncInstruction()
256 CHECK(!async_computation->IsCustomCallComputation()); in HloAsyncInstruction()
257 CHECK(!async_computation->IsFusionComputation()); in HloAsyncInstruction()
258 async_computation->AddAsyncInstruction(this); in HloAsyncInstruction()
264 HloComputation* async_computation, std::optional<int64_t> async_group_id, in HloAsyncInstruction() argument
270 AppendComputation(async_computation); in HloAsyncInstruction()
271 CHECK(!async_computation->IsCustomCallComputation()); in HloAsyncInstruction()
272 CHECK(!async_computation->IsFusionComputation()); in HloAsyncInstruction()
273 async_computation->AddAsyncInstruction(this); in HloAsyncInstruction()
|
| H A D | hlo_instruction_test.cc | 1717 std::unique_ptr<HloComputation> async_computation = async_builder.Build(); in TEST_F() local
1724 s_tuple, {entry_param}, async_computation.get(), in TEST_F()
1729 s_tuple, async_start, async_computation.get(), in TEST_F()
1733 s2, async_update, async_computation.get(), in TEST_F()
1739 module->AddEmbeddedComputation(std::move(async_computation)); in TEST_F()
|
| H A D | hlo_computation.cc | 773 HloComputation* async_computation = in CreateAsyncInstructions() local
782 async_computation, /*async_group_id=*/std::nullopt, in CreateAsyncInstructions()
785 root->shape(), async_start, async_computation, in CreateAsyncInstructions()
|
| H A D | hlo_instruction.h | 631 HloComputation* async_computation,
636 HloComputation* async_computation,
641 HloComputation* async_computation,
|
| H A D | hlo_instruction.cc | 1189 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncStart() argument
1192 HloOpcode::kAsyncStart, shape, operands, async_computation, in CreateAsyncStart()
1198 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncUpdate() argument
1201 HloOpcode::kAsyncUpdate, shape, operand, async_computation, in CreateAsyncUpdate()
1207 HloComputation* async_computation, std::optional<int64_t> async_group_id, in CreateAsyncDone() argument
1210 HloOpcode::kAsyncDone, shape, operand, async_computation, async_group_id, in CreateAsyncDone()
|
| H A D | hlo_parser.cc | 1610 std::optional<HloComputation*> async_computation; in CreateInstruction() local
1674 async_computation = computations_.back().get(); in CreateInstruction()
1678 &async_computation}; in CreateInstruction()
1685 *shape, operands, *async_computation, async_group_id, in CreateInstruction()
1690 *shape, operands[0], *async_computation, async_group_id, in CreateInstruction()
1694 *shape, operands[0], *async_computation, async_group_id, in CreateInstruction()
|
| H A D | hlo_instructions.h | 216 HloComputation* async_computation,
221 HloComputation* async_computation,
|