| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | async_op_canonicalizer_test.cc | 66 HloInstruction* async_done = FindInstruction(module.get(), "async-done"); in TEST_F() local
70 EXPECT_EQ(async_done->async_group_id(), 0); in TEST_F()
74 async_done->async_wrapped_computation()); in TEST_F()
129 HloInstruction* async_done = FindInstruction(module.get(), "async-done"); in TEST_F() local
133 EXPECT_EQ(async_done->async_group_id(), 0); in TEST_F()
137 async_done->async_wrapped_computation()); in TEST_F()
|
| D | async_collective_creator.cc | 158 TF_ASSIGN_OR_RETURN(HloInstruction * async_done, in Run()
162 HloInstruction* async_start = async_done->mutable_operand(0); in Run()
163 replaced_pairs[ata] = ReplacedAsync{async_start, async_done}; in Run()
|
| D | dfs_hlo_visitor_with_default.h | 235 Status HandleAsyncDone(HloInstructionPtr async_done) override { in HandleAsyncDone() argument
236 return DefaultAction(async_done); in HandleAsyncDone()
|
| D | hlo_verifier.cc | 1486 Status ShapeVerifier::HandleAsyncDone(HloInstruction* async_done) { in HandleAsyncDone() argument
1487 TF_RETURN_IF_ERROR(CheckAsyncOpComputationThreadName(async_done)); in HandleAsyncDone()
1489 async_done, async_done->operand(0)->shape())); in HandleAsyncDone()
1490 const Shape& root_shape = async_done->operand(0)->shape().tuple_shapes(1); in HandleAsyncDone()
1491 if (root_shape != async_done->shape()) { in HandleAsyncDone()
1495 HloOpcodeString(async_done->opcode()), async_done->shape().ToString(), in HandleAsyncDone()
1498 return CheckAsyncOpOperand(async_done); in HandleAsyncDone()
|
| D | tuple_points_to_analysis.cc | 363 Status TuplePointsToAnalysis::HandleAsyncDone(HloInstruction* async_done) { in HandleAsyncDone() argument
365 PointsToSet& points_to_set = CreateEmptyPointsToSet(async_done); in HandleAsyncDone()
367 GetPointsToSet(async_done->operand(0)); in HandleAsyncDone()
|
| D | hlo_computation.cc | 784 HloInstruction* async_done = AddInstruction(HloInstruction::CreateAsyncDone( in CreateAsyncInstructions() local
789 async_done->set_metadata(instruction->metadata()); in CreateAsyncInstructions()
790 async_done->CopyBackendConfigFrom(instruction); in CreateAsyncInstructions()
791 TF_RETURN_IF_ERROR(ReplaceInstruction(instruction, async_done)); in CreateAsyncInstructions()
792 return async_done; in CreateAsyncInstructions()
|
| D | hlo_dataflow_analysis.cc | 688 bool HloDataflowAnalysis::UpdateAsyncDoneValueSet(HloInstruction* async_done) { in UpdateAsyncDoneValueSet() argument
689 CHECK_EQ(async_done->opcode(), HloOpcode::kAsyncDone); in UpdateAsyncDoneValueSet()
692 async_done->async_wrapped_computation()->root_instruction(); in UpdateAsyncDoneValueSet()
696 async_done->operand(0)->shape(), in UpdateAsyncDoneValueSet()
702 GetValueSet(async_done->operand(0), index); in UpdateAsyncDoneValueSet()
705 HloValueSet& value_set = GetValueSet(async_done, output_index); in UpdateAsyncDoneValueSet()
|
| D | hlo_dataflow_analysis.h | 272 bool UpdateAsyncDoneValueSet(HloInstruction* async_done);
|
| D | hlo_module_test.cc | 625 auto* async_done, in TEST_F()
628 auto* parallel_thread_computation = async_done->async_wrapped_computation(); in TEST_F()
|
| D | tuple_points_to_analysis.h | 251 Status HandleAsyncDone(HloInstruction* async_done) override;
|
| D | hlo_cost_analysis.h | 102 Status HandleAsyncDone(const HloInstruction* async_done) override;
|
| D | hlo_evaluator.h | 305 Status HandleAsyncDone(HloInstruction* async_done) override;
|
| D | hlo_instruction_test.cc | 792 auto* async_done, in TEST_F()
795 auto* async_start = async_done->operand(0); in TEST_F()
799 EXPECT_EQ(async_done->async_execution_thread(), "parallel_thread"); in TEST_F()
804 EXPECT_EQ(async_done->async_wrapped_computation()->execution_thread(), in TEST_F()
809 EXPECT_EQ(computation->root_instruction(), async_done); in TEST_F()
|
| D | hlo_verifier.h | 202 Status HandleAsyncDone(HloInstruction* async_done) override;
|
| D | hlo_dataflow_analysis_test.cc | 1094 const HloInstruction* async_done = in TEST_P() local
1121 EXPECT_FALSE(analysis.ValueIsDefinedAt(async_done, /*index=*/{})); in TEST_P()
1122 EXPECT_THAT(HloValuesAt(async_done, /*index=*/{}), in TEST_P()
1159 const HloInstruction* async_done = in TEST_P() local
1180 EXPECT_THAT(HloValuesAt(async_done), in TEST_P()
|
| D | hlo_evaluator.cc | 3074 Status HloEvaluator::HandleAsyncDone(HloInstruction* async_done) { in HandleAsyncDone() argument
3076 GetEvaluatedLiteralFor(async_done->operand(0)); in HandleAsyncDone()
3077 evaluated_[async_done] = Literal(async_done->shape()); in HandleAsyncDone()
3078 TF_RETURN_IF_ERROR(evaluated_[async_done].CopyFrom(operand_tuple_literal, in HandleAsyncDone()
|