/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_module_group_util.cc | 108 HloInstruction* recv_done = in GlobalPredecessors() local 109 metadata_.GetChannel(*instruction->channel_id()).recv_done; in GlobalPredecessors() 110 CHECK(recv_done->opcode() == HloOpcode::kRecvDone); in GlobalPredecessors() 111 CHECK_EQ(recv_done->operand_count(), 1); in GlobalPredecessors() 112 HloInstruction* recv = recv_done->mutable_operand(0); in GlobalPredecessors() 179 const HloInstruction* recv_done = instruction->users().front(); in GlobalSuccessors() local 180 CHECK(recv_done->opcode() == HloOpcode::kRecvDone); in GlobalSuccessors() 188 HloInstruction* recv_done = in GlobalSuccessors() local 189 metadata_.GetChannel(*instruction->channel_id()).recv_done; in GlobalSuccessors() 190 add_unique_successor(recv_done); in GlobalSuccessors()
|
D | hlo_module_group_metadata.cc | 381 TF_RET_CHECK(channel.recv_done == nullptr) in RecordInstructions() 384 channel.recv_done = hlo; in RecordInstructions() 457 if (channel.recv_done == nullptr) { in VerifyChannelInstructions() 466 ShapeUtil::GetTupleElementShape(channel.recv_done->shape(), 0); in VerifyChannelInstructions() 487 auto recv_done_device = GetInstructionDevice(*channel.recv_done); in VerifyChannelInstructions() 490 channel.recv_done->ToString()); in VerifyChannelInstructions() 509 TF_RETURN_IF_ERROR(CheckCommunicatingInstruction(channel.recv_done)); in VerifyChannelInstructions() 519 path, GetCompanionsPath(channel.recv_done))) { in VerifyChannelInstructions() 547 << " recv_done=" << channel.recv_done->name(); in DumpCollectedStats()
|
D | logical_buffer_analysis.cc | 144 Status LogicalBufferAnalysis::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument 148 NewLogicalBuffer(recv_done, /*index=*/{}); in HandleRecvDone() 149 NewLogicalBuffer(recv_done, /*index=*/{1}); in HandleRecvDone()
|
D | hlo_reachability_test.cc | 199 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_F() local 202 auto computation = module->AddEntryComputation(builder.Build(recv_done)); in TEST_F() 204 EXPECT_TRUE(reachability->IsReachable(param, recv_done)); in TEST_F()
|
D | tuple_points_to_analysis.cc | 290 Status TuplePointsToAnalysis::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument 293 PointsToSet& points_to_set = CreateEmptyPointsToSet(recv_done); in HandleRecvDone() 295 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{}), in HandleRecvDone() 298 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{1}), in HandleRecvDone() 302 GetPointsToSet(recv_done->operand(0)); in HandleRecvDone()
|
D | logical_buffer_analysis.h | 67 Status HandleRecvDone(HloInstruction* recv_done) override;
|
D | dfs_hlo_visitor_with_default.h | 225 Status HandleRecvDone(HloInstructionPtr recv_done) override { in HandleRecvDone() argument 226 return DefaultAction(recv_done); in HandleRecvDone()
|
D | hlo_dataflow_analysis.h | 225 bool UpdateRecvDoneValueSet(HloInstruction* recv_done);
|
D | hlo_module_group_metadata.h | 110 HloInstruction* recv_done = nullptr; member
|
D | hlo_verifier.cc | 1029 Status ShapeVerifier::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument 1031 recv_done, in HandleRecvDone() 1033 {ShapeUtil::GetTupleElementShape(recv_done->operand(0)->shape(), 0), in HandleRecvDone() 1560 const HloInstruction* recv_done = instruction->users().front(); in VerifyChannels() local 1561 TF_RET_CHECK(recv_done->opcode() == HloOpcode::kRecvDone); in VerifyChannels() 1562 TF_RETURN_IF_ERROR(CheckSameChannel(instruction, recv_done)); in VerifyChannels() 1563 TF_RETURN_IF_ERROR(CheckSameIsHostTransfer(instruction, recv_done)); in VerifyChannels()
|
D | tuple_points_to_analysis_test.cc | 392 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_F() local 398 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(recv_done).IsAmbiguous()); in TEST_F() 399 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(recv_done).IsDistinct()); in TEST_F() 403 ExpectHasBufferAliases(recv, {0}, {{recv, {0}}, {recv_done, {0}}}); in TEST_F()
|
D | hlo_dataflow_analysis.cc | 479 bool HloDataflowAnalysis::UpdateRecvDoneValueSet(HloInstruction* recv_done) { in UpdateRecvDoneValueSet() argument 480 CHECK_EQ(recv_done->opcode(), HloOpcode::kRecvDone); in UpdateRecvDoneValueSet() 483 for (auto& pair : GetInstructionValueSet(recv_done)) { in UpdateRecvDoneValueSet() 492 GetValueSet(recv_done->operand(0), index); in UpdateRecvDoneValueSet()
|
D | tuple_points_to_analysis.h | 255 Status HandleRecvDone(HloInstruction* recv_done) override;
|
D | hlo_cost_analysis.h | 70 Status HandleRecvDone(const HloInstruction* recv_done) override;
|
D | hlo_verifier.h | 103 Status HandleRecvDone(HloInstruction* recv_done) override;
|
D | dfs_hlo_visitor.h | 275 virtual Status HandleRecvDone(HloInstructionPtr recv_done) = 0;
|
D | hlo_dataflow_analysis_test.cc | 1315 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_P() local 1328 EXPECT_TRUE(analysis.ValueIsDefinedAt(recv_done, /*index=*/{})); in TEST_P() 1329 EXPECT_FALSE(analysis.ValueIsDefinedAt(recv_done, /*index=*/{0})); in TEST_P() 1330 EXPECT_TRUE(analysis.ValueIsDefinedAt(recv_done, /*index=*/{1})); in TEST_P() 1331 EXPECT_THAT(HloValuesAt(recv_done, /*index=*/{0}), in TEST_P()
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
D | rpc_rendezvous_mgr.cc | 102 void Start(std::function<void()> recv_done) override { in Start() argument 103 StartRTCall(std::move(recv_done)); in Start() 138 void StartRTCall(std::function<void()> recv_done) { in StartRTCall() argument 142 recv_done = std::move(recv_done)](const Status& s) { in StartRTCall() 150 recv_done(); in StartRTCall()
|
/external/tensorflow/tensorflow/core/framework/ |
D | rendezvous_test.cc | 306 auto recv_done = [this, &state, i](const Status& status, in TEST_F() local 324 SchedClosure([this, i, micros, recv_done]() { in TEST_F() 327 recv_done); in TEST_F()
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | base_rendezvous_mgr.h | 227 virtual void Start(std::function<void()> recv_done) = 0;
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter.h | 89 Status HandleRecvDone(HloInstruction* recv_done) override;
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | ir_emitter.h | 158 Status HandleRecvDone(HloInstruction* recv_done) override;
|
D | ir_emitter.cc | 1999 Status IrEmitter::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument
|