| /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()
180 const HloInstruction* recv_done = instruction->users().front(); in GlobalSuccessors() local
181 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 | 364 TF_RET_CHECK(channel.recv_done == nullptr) in RecordInstructions()
367 channel.recv_done = hlo; in RecordInstructions()
439 if (channel.recv_done == nullptr) { in VerifyChannelInstructions()
448 ShapeUtil::GetTupleElementShape(channel.recv_done->shape(), 0); in VerifyChannelInstructions()
469 auto recv_done_device = GetInstructionDevice(*channel.recv_done); in VerifyChannelInstructions()
472 channel.recv_done->ToString()); in VerifyChannelInstructions()
491 TF_RETURN_IF_ERROR(CheckCommunicatingInstruction(channel.recv_done)); in VerifyChannelInstructions()
501 path, GetCompanionsPath(channel.recv_done))) { in VerifyChannelInstructions()
529 << " recv_done=" << channel.recv_done->name(); in DumpCollectedStats()
|
| D | logical_buffer_analysis.cc | 142 Status LogicalBufferAnalysis::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument
146 NewLogicalBuffer(recv_done, /*index=*/{}); in HandleRecvDone()
147 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 | 287 Status TuplePointsToAnalysis::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument
290 PointsToSet& points_to_set = CreateEmptyPointsToSet(recv_done); in HandleRecvDone()
292 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{}), in HandleRecvDone()
295 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{1}), in HandleRecvDone()
299 GetPointsToSet(recv_done->operand(0)); in HandleRecvDone()
|
| D | dfs_hlo_visitor_with_default.h | 197 Status HandleRecvDone(HloInstructionPtr recv_done) override { in HandleRecvDone() argument
198 return DefaultAction(recv_done); in HandleRecvDone()
|
| D | hlo_verifier.cc | 713 Status ShapeVerifier::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument
715 recv_done, in HandleRecvDone()
717 {ShapeUtil::GetTupleElementShape(recv_done->operand(0)->shape(), 0), in HandleRecvDone()
1156 const HloInstruction* recv_done = instruction->users().front(); in VerifySendsAndRecvs() local
1157 TF_RET_CHECK(recv_done->opcode() == HloOpcode::kRecvDone); in VerifySendsAndRecvs()
1158 TF_RETURN_IF_ERROR(CheckSameChannel(instruction, recv_done)); in VerifySendsAndRecvs()
1159 TF_RETURN_IF_ERROR(CheckSameIsHostTransfer(instruction, recv_done)); in VerifySendsAndRecvs()
|
| D | logical_buffer_analysis.h | 64 Status HandleRecvDone(HloInstruction* recv_done) override;
|
| D | hlo_dataflow_analysis.cc | 340 bool HloDataflowAnalysis::UpdateRecvDoneValueSet(HloInstruction* recv_done) { in UpdateRecvDoneValueSet() argument
341 CHECK_EQ(recv_done->opcode(), HloOpcode::kRecvDone); in UpdateRecvDoneValueSet()
344 for (auto& pair : GetInstructionValueSet(recv_done)) { in UpdateRecvDoneValueSet()
353 GetValueSet(recv_done->operand(0), index); in UpdateRecvDoneValueSet()
|
| D | hlo_dataflow_analysis.h | 191 bool UpdateRecvDoneValueSet(HloInstruction* recv_done);
|
| D | hlo_module_group_metadata.h | 110 HloInstruction* recv_done = nullptr; member
|
| D | buffer_liveness_test.cc | 334 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_F() local
336 HloInstruction::CreateSend(recv_done, token, /*channel_id=*/1)); in TEST_F()
344 {param, add, token, recv, recv_done, send, send_done}); in TEST_F()
|
| D | hlo_cost_analysis.h | 67 Status HandleRecvDone(const HloInstruction* recv_done) override;
|
| D | tuple_points_to_analysis.h | 252 Status HandleRecvDone(HloInstruction* recv_done) override;
|
| D | tuple_points_to_analysis_test.cc | 364 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_F() local
370 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(recv_done).IsAmbiguous()); in TEST_F()
371 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(recv_done).IsDistinct()); in TEST_F()
375 ExpectHasBufferAliases(recv, {0}, {{recv, {0}}, {recv_done, {0}}}); in TEST_F()
|
| D | hlo_verifier.h | 92 Status HandleRecvDone(HloInstruction* recv_done) override;
|
| D | dfs_hlo_visitor.h | 255 virtual Status HandleRecvDone(HloInstructionPtr recv_done) = 0;
|
| D | hlo_dataflow_analysis_test.cc | 1207 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); in TEST_P() local
1220 EXPECT_TRUE(analysis.ValueIsDefinedAt(recv_done, /*index=*/{})); in TEST_P()
1221 EXPECT_FALSE(analysis.ValueIsDefinedAt(recv_done, /*index=*/{0})); in TEST_P()
1222 EXPECT_TRUE(analysis.ValueIsDefinedAt(recv_done, /*index=*/{1})); in TEST_P()
1223 EXPECT_THAT(HloValuesAt(recv_done, /*index=*/{0}), in TEST_P()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 56 void Start(std::function<void()> recv_done) override { in Start() argument
59 StatusCallback cb = [this, recv_done](const Status& s) { in Start()
67 [this, recv_done](const Status& s) { in Start()
72 recv_done(); in Start()
80 recv_done(); in Start()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 101 void Start(std::function<void()> recv_done) override { in Start() argument
102 StartRTCall(std::move(recv_done)); in Start()
137 void StartRTCall(std::function<void()> recv_done) { in StartRTCall() argument
141 [this](std::function<void()> recv_done, in StartRTCall()
148 recv_done(); in StartRTCall()
150 std::move(recv_done), _1); in StartRTCall()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous_test.cc | 184 auto recv_done = [this, &state, i](const Status& status, in TEST_F() local
202 SchedClosure([this, i, micros, recv_done]() { in TEST_F()
205 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 | 90 Status HandleRecvDone(HloInstruction* recv_done) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 172 Status HandleRecvDone(HloInstruction* recv_done) override;
|
| D | ir_emitter.cc | 2085 Status IrEmitter::HandleRecvDone(HloInstruction* recv_done) { in HandleRecvDone() argument
|