| /external/tensorflow/tensorflow/core/framework/ |
| D | local_rendezvous.cc | 34 enum Type { kSend = 0, kRecv = 1 }; enumerator
44 : Item(recv_args, kRecv) { in Item()
161 DCHECK_EQ(item->type, Item::kRecv); in Send()
183 if (queue->head == nullptr || queue->head->type == Item::kRecv) { in RecvAsync()
212 if (queue->head != nullptr && queue->head->type == Item::kRecv) { in RecvAsync()
325 if (item->type == Item::kRecv) { in StartAbort()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_utils.cc | 64 position.instruction->opcode() == HloOpcode::kRecv)) { in IsValueAllowedInAlternateMemory()
|
| D | hlo_module_group_metadata.cc | 202 case HloOpcode::kRecv: in IsChannelInstruction()
243 case HloOpcode::kRecv: in PeerComputation()
368 if (hlo->opcode() == HloOpcode::kRecv) { in RecordInstructions()
|
| D | hlo_query.cc | 159 } else if (hlo->opcode() == HloOpcode::kRecv) { in HasX64TransformedHostTransfer()
|
| D | hlo_opcode.h | 118 V(kRecv, "recv", 1) \
|
| D | hlo_module_group_util.cc | 176 if (instruction->opcode() == HloOpcode::kRecv && in GlobalSuccessors()
|
| D | hlo_verifier.cc | 1093 case HloOpcode::kRecv: in CheckMixedPrecisionOperands()
1206 case HloOpcode::kRecv: in CheckShape()
1558 case HloOpcode::kRecv: { in VerifyChannels()
1572 TF_RET_CHECK(instruction->operand(0)->opcode() == HloOpcode::kRecv); in VerifyChannels()
|
| D | while_loop_simplifier.cc | 1214 HloOpcode::kRecv, HloOpcode::kRecvDone}) || in Run()
1217 HloOpcode::kRecv, HloOpcode::kRecvDone})) { in Run()
|
| D | instruction_fusion.cc | 170 case HloOpcode::kRecv: in IsExpensive()
|
| D | layout_assignment.cc | 412 instruction->opcode() == HloOpcode::kRecv) { in BuildHostChannelConstraints()
491 instruction->opcode() == HloOpcode::kRecv) { in AddMandatoryConstraints()
2302 case HloOpcode::kRecv: in InstructionCanChangeLayout()
|
| D | hlo_instruction.cc | 217 case HloOpcode::kRecv: in CreateFromProto()
1526 case HloOpcode::kRecv: in HasSideEffectNoRecurse()
1668 case HloOpcode::kRecv: in CloneWithNewOperands()
2206 case HloOpcode::kRecv: in IdenticalSlowPath()
3244 case HloOpcode::kRecv: in Visit()
|
| D | sharding_propagation.cc | 261 case HloOpcode::kRecv: in PickRepresentativeOperand()
1591 } else if (opcode == HloOpcode::kSend || opcode == HloOpcode::kRecv || in CheckAndUpdateDeviceAssignmentsInWhileBody()
|
| D | hlo_graph_dumper.cc | 1073 case HloOpcode::kRecv: in GetInstructionColor()
|
| D | hlo_parser.cc | 174 case HloOpcode::kRecv: in CanInferShape()
1454 case HloOpcode::kRecv: { in ParseInstructionRhs()
|
| D | hlo_instructions.cc | 534 HloOpcode::kRecv, in HloRecvInstruction()
|
| D | memory_space_assignment_test.cc | 2382 instruction->opcode() == HloOpcode::kRecv) { in TEST_P()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 39 constexpr char kRecv[] = "_Recv"; variable
486 NodeSetUp("Recv1", kRecv, kCPU0, 2003, &recv1); in TEST_F()
487 NodeSetUp("Recv2", kRecv, kCPU1, 2004, &recv2); in TEST_F()
562 NodeSetUp("Node9", kRecv, kCPU0, 1000, &node9); in TEST_F()
564 NodeSetUp("Node11", kRecv, kCPU0, 999, &node11); in TEST_F()
578 EXPECT_EQ(manager.GetCurrNode()->op(), kRecv); in TEST_F()
596 EXPECT_EQ(manager.GetCurrNode()->op(), kRecv); in TEST_F()
618 EXPECT_EQ(manager.GetCurrNode()->op(), kRecv); in TEST_F()
2974 if (op == kRecv) { in TEST_F()
2983 EXPECT_EQ(op_count.at(kSend), op_count.at(kRecv)); in TEST_F()
[all …]
|
| D | op_level_cost_estimator.cc | 66 constexpr char kRecv[] = "_Recv"; variable
455 device_cost_impl_.emplace(kRecv, in OpLevelCostEstimator()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 328 return xla::HloOpcode::kRecv; in MhloToHloOpcode()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 365 case HloOpcode::kRecv: in IsConstantVisitor()
3226 HloOpcode::kRecv, {token})); in RecvWithToken()
3313 HloOpcode::kRecv, {token})); in RecvFromHost()
|