| /external/tensorflow/tensorflow/compiler/xla/service/ |
| buffer_liveness_test.cc | 332 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); local
334 HloInstruction::CreateSend(recv_done, /*channel_id=*/1));
342 recv_done, send, send_done};
[all...] |
| tuple_points_to_analysis_test.cc | 347 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); local
353 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(recv_done).IsAmbiguous());
354 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(recv_done).IsDistinct());
358 ExpectHasBufferAliases(recv, {0}, {{recv, {0}}, {recv_done, {}}});
[all...] |
| hlo_verifier.cc | 324 const HloInstruction* recv_done = recv->users().front(); local
325 TF_RET_CHECK(recv_done->opcode() == HloOpcode::kRecvDone);
326 TF_RETURN_IF_ERROR(CheckSameChannel(recv, recv_done));
329 {recv_done->shape(), ShapeUtil::MakeShape(U32, {})}));
332 Status ShapeVerifier::HandleRecvDone(HloInstruction* recv_done) {
333 TF_RET_CHECK(recv_done->operands().size() == 1);
334 const HloInstruction* recv = recv_done->operand(0);
336 TF_RETURN_IF_ERROR(CheckSameChannel(recv, recv_done));
337 return CheckShape(recv_done, recv->shape().tuple_shapes(0));
[all...] |
| hlo_dataflow_analysis_test.cc | 1186 auto recv_done = builder.AddInstruction(HloInstruction::CreateRecvDone(recv)); local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| rendezvous_test.cc | 184 auto recv_done = [this, &state, i](const Status& status, local
202 SchedClosure([this, i, micros, recv_done]() {
205 recv_done);
|
