Searched refs:RecvOp (Results 1 – 6 of 6) sorted by relevance
53 class RecvOp : public XlaOpKernel { class55 explicit RecvOp(OpKernelConstruction* ctx);62 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);65 RecvOp::RecvOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in RecvOp() function in tensorflow::__anon1ab2458c0111::RecvOp75 void RecvOp::Compile(XlaOpKernelContext* ctx) { in Compile()82 REGISTER_XLA_OP(Name("XlaRecv"), RecvOp);
120 RecvOp::RecvOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in RecvOp() function in tensorflow::RecvOp163 void RecvOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()195 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_CPU), RecvOp);196 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_DEFAULT), RecvOp);199 REGISTER_KERNEL_BUILDER(Name("Recv").Device(DEVICE_CPU), RecvOp);200 REGISTER_KERNEL_BUILDER(Name("Recv").Device(DEVICE_DEFAULT), RecvOp);203 Name("_HostRecv").Device(DEVICE_DEFAULT).HostMemory("tensor"), RecvOp);209 REGISTER_KERNEL_BUILDER(Name("_HostRecv").Device(DEVICE_CPU), RecvOp); in InitModule()
37 class RecvOp : public AsyncOpKernel {39 explicit RecvOp(OpKernelConstruction* ctx);47 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
74 friend class RecvOp; member
710 static LogicalResult Verify(RecvOp op) { in Verify()
615 LogicalResult ExportXlaOp(RecvOp op, OpLoweringContext ctx) { in ExportXlaOp()