Home
last modified time | relevance | path

Searched refs:recv_done (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_util.cc108 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()
Dhlo_module_group_metadata.cc381 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()
Dlogical_buffer_analysis.cc144 Status LogicalBufferAnalysis::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument
148 NewLogicalBuffer(recv_done, /*index=*/{}); in HandleRecvDone()
149 NewLogicalBuffer(recv_done, /*index=*/{1}); in HandleRecvDone()
Dhlo_reachability_test.cc199 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()
Dtuple_points_to_analysis.cc290 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()
Dlogical_buffer_analysis.h67 Status HandleRecvDone(HloInstruction* recv_done) override;
Ddfs_hlo_visitor_with_default.h225 Status HandleRecvDone(HloInstructionPtr recv_done) override { in HandleRecvDone() argument
226 return DefaultAction(recv_done); in HandleRecvDone()
Dhlo_dataflow_analysis.h225 bool UpdateRecvDoneValueSet(HloInstruction* recv_done);
Dhlo_module_group_metadata.h110 HloInstruction* recv_done = nullptr; member
Dhlo_verifier.cc1029 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()
Dtuple_points_to_analysis_test.cc392 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()
Dhlo_dataflow_analysis.cc479 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()
Dtuple_points_to_analysis.h255 Status HandleRecvDone(HloInstruction* recv_done) override;
Dhlo_cost_analysis.h70 Status HandleRecvDone(const HloInstruction* recv_done) override;
Dhlo_verifier.h103 Status HandleRecvDone(HloInstruction* recv_done) override;
Ddfs_hlo_visitor.h275 virtual Status HandleRecvDone(HloInstructionPtr recv_done) = 0;
Dhlo_dataflow_analysis_test.cc1315 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/
Drpc_rendezvous_mgr.cc102 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/
Drendezvous_test.cc306 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/
Dbase_rendezvous_mgr.h227 virtual void Start(std::function<void()> recv_done) = 0;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h89 Status HandleRecvDone(HloInstruction* recv_done) override;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h158 Status HandleRecvDone(HloInstruction* recv_done) override;
Dir_emitter.cc1999 Status IrEmitter::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument