/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);
|