Home
last modified time | relevance | path

Searched refs:RecvOp (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsendrecv_ops.cc53 class RecvOp : public XlaOpKernel { class
55 explicit RecvOp(OpKernelConstruction* ctx);
62 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
65 RecvOp::RecvOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in RecvOp() function in tensorflow::__anon1ab2458c0111::RecvOp
75 void RecvOp::Compile(XlaOpKernelContext* ctx) { in Compile()
82 REGISTER_XLA_OP(Name("XlaRecv"), RecvOp);
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc120 RecvOp::RecvOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in RecvOp() function in tensorflow::RecvOp
163 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()
Dsendrecv_ops.h37 class RecvOp : public AsyncOpKernel {
39 explicit RecvOp(OpKernelConstruction* ctx);
47 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
/external/tensorflow/tensorflow/core/framework/
Drendezvous.h74 friend class RecvOp; member
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dhlo_ops.cc710 static LogicalResult Verify(RecvOp op) { in Verify()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc615 LogicalResult ExportXlaOp(RecvOp op, OpLoweringContext ctx) { in ExportXlaOp()