HomeSort by relevance Sort by last modified time
    Searched refs:RecvOp (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
sendrecv_ops.h 37 class RecvOp : public AsyncOpKernel {
39 explicit RecvOp(OpKernelConstruction* ctx);
47 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
sendrecv_ops.cc 123 RecvOp::RecvOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) {
171 void RecvOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) {
196 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_CPU), RecvOp);
197 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_GPU), RecvOp);
200 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_SYCL), RecvOp);
203 REGISTER_KERNEL_BUILDER(Name("_HostRecv").Device(DEVICE_CPU), RecvOp);
205 Name("_HostRecv").Device(DEVICE_GPU).HostMemory("tensor"), RecvOp);
209 Name("_HostRecv").Device(DEVICE_SYCL).HostMemory("tensor"), RecvOp);
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
sendrecv_ops.cc 53 class RecvOp : public XlaOpKernel {
55 explicit RecvOp(OpKernelConstruction* ctx);
62 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
65 RecvOp::RecvOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {
75 void RecvOp::Compile(XlaOpKernelContext* ctx) {
82 REGISTER_XLA_OP(Name("_XLARecv"), RecvOp);
  /external/tensorflow/tensorflow/compiler/jit/
xla_device_ops.h 52 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE), RecvOp); \
56 Name("_HostRecv").Device(DEVICE).HostMemory("tensor"), RecvOp); \
  /external/tensorflow/tensorflow/core/framework/
rendezvous.h 79 friend class RecvOp;

Completed in 206 milliseconds