Home
last modified time | relevance | path

Searched refs:kRecv (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc88 const string kRecv = "_Recv"; member in tensorflow::grappler::VirtualSchedulerTest
2121 NodeSetUp("Recv1", kRecv, kCPU0, 2003, &recv1); in TEST_F()
2122 NodeSetUp("Recv2", kRecv, kCPU1, 2004, &recv2); in TEST_F()
2197 NodeSetUp("Node9", kRecv, kCPU0, 1000, &node9); in TEST_F()
2199 NodeSetUp("Node11", kRecv, kCPU0, 999, &node11); in TEST_F()
2213 EXPECT_EQ(kRecv, manager.GetCurrNode()->op()); in TEST_F()
2231 EXPECT_EQ(kRecv, manager.GetCurrNode()->op()); in TEST_F()
2253 EXPECT_EQ(kRecv, manager.GetCurrNode()->op()); in TEST_F()
2798 if (op == kRecv) { in TEST_F()
2807 EXPECT_EQ(op_count.at(kSend), op_count.at(kRecv)); in TEST_F()
[all …]
Dop_level_cost_estimator.cc50 constexpr char kRecv[] = "_Recv"; variable
268 {kRecv, wrap(&OpLevelCostEstimator::PredictIdentity)}, in OpLevelCostEstimator()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_metadata.cc185 case HloOpcode::kRecv: in IsChannelInstruction()
226 case HloOpcode::kRecv: in PeerComputation()
351 if (hlo->opcode() == HloOpcode::kRecv) { in RecordInstructions()
Dhlo_opcode.h113 V(kRecv, "recv", 1) \
Dhlo_verifier.cc773 case HloOpcode::kRecv: in CheckMixedPrecisionOperands()
874 case HloOpcode::kRecv: in CheckShape()
1153 case HloOpcode::kRecv: { in VerifySendsAndRecvs()
1168 TF_RET_CHECK(instruction->operand(0)->opcode() == HloOpcode::kRecv); in VerifySendsAndRecvs()
Dhlo_module_group_util.cc177 if (instruction->opcode() == HloOpcode::kRecv && in GlobalSuccessors()
Dwhile_loop_simplifier.cc1002 HloOpcode::kRecv, HloOpcode::kRecvDone}) || in Run()
1005 HloOpcode::kRecv, HloOpcode::kRecvDone})) { in Run()
Dinstruction_fusion.cc144 case HloOpcode::kRecv: in IsExpensive()
Dlayout_assignment.cc402 instruction->opcode() == HloOpcode::kRecv) { in BuildHostChannelConstraints()
477 instruction->opcode() == HloOpcode::kRecv) { in AddMandatoryConstraints()
2064 case HloOpcode::kRecv: in InstructionCanChangeLayout()
Dhlo_instruction.cc195 case HloOpcode::kRecv: in CreateFromProto()
1241 case HloOpcode::kRecv: in HasSideEffectNoRecurse()
1356 case HloOpcode::kRecv: in CloneWithNewOperands()
1806 case HloOpcode::kRecv: in IdenticalSlowPath()
2641 case HloOpcode::kRecv: in Visit()
Dhlo.proto124 // Only present for kSend or kRecv.
Dhlo_graph_dumper.cc1045 case HloOpcode::kRecv: in GetInstructionColor()
Dhlo_instructions.cc440 HloOpcode::kRecv, in HloRecvInstruction()
Dhlo_parser.cc957 case HloOpcode::kRecv: { in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc244 case HloOpcode::kRecv: in IsConstantVisitor()
2394 HloOpcode::kRecv, {token})); in RecvWithToken()
2481 HloOpcode::kRecv, {token})); in RecvFromHost()