Searched defs:recv_op (Results 1 – 7 of 7) sorted by relevance
78 } else if (auto recv_op = llvm::dyn_cast_or_null<mlir::TF::DTensorRecv>(op)) { in BackwardShapeInferenceToRestoreOp() local
69 auto recv_op = GetCorrespondingDTensorSendRecvOp<mlir::TF::DTensorSend>( in LowerDTensorSendRecvsOps() local
216 auto recv_op = builder.create<mlir::TF::DTensorRecv>( in LowerToSendRecv() local
294 auto recv_op = builder.create<mlir::TF::DTensorRecv>( in CloneEmptyIfWithPredicate() local
63 RecvTPUEmbeddingActivationsOp recv_op; in RunOnRegion() local
524 const NodeDef* recv_op = it->second; in Init() local
299 const string recv_op = (host_memory) ? "_HostRecv" : "_Recv"; in AddRecv() local