Searched defs:RecvOp (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsendrecv_ops.h37 class RecvOp : public AsyncOpKernel { class in namespace:tensorflow
39 explicit RecvOp(OpKernelConstruction* ctx);
47 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
H A Dsendrecv_ops.cc123 RecvOp::RecvOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { function in class:tensorflow::RecvOp
171 void RecvOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) {
196 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_CPU), RecvOp); variable
197 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_GPU), RecvOp); variable
200 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_SYCL), RecvOp); variable
203 REGISTER_KERNEL_BUILDER(Name("_HostRecv").Device(DEVICE_CPU), RecvOp); variable
205 Name("_HostRecv").Device(DEVICE_GPU).HostMemory("tensor"), RecvOp);
209 Name("_HostRecv").Device(DEVICE_SYCL).HostMemory("tensor"), RecvOp);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dsendrecv_ops.cc53 class RecvOp : public XlaOpKernel { class in namespace:tensorflow::__anon25453
55 explicit RecvOp(OpKernelConstruction* ctx);
62 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
65 RecvOp::RecvOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { function in class:tensorflow::__anon25453::RecvOp
75 void RecvOp::Compile(XlaOpKernelContext* ctx) {
82 REGISTER_XLA_OP(Name("_XLARecv"), RecvOp); member in namespace:tensorflow::__anon25453

Completed in 5299 milliseconds