Home
last modified time | relevance | path

Searched refs:RecvOp (Results 1 – 8 of 8) 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::__anon233ac8c10111::RecvOp
75 void RecvOp::Compile(XlaOpKernelContext* ctx) { in Compile()
82 REGISTER_XLA_OP(Name("XlaRecv"), RecvOp);
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc137 RecvOp::RecvOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in RecvOp() function in tensorflow::RecvOp
159 string RecvOp::TraceString(const OpKernelContext& ctx, bool verbose) const { in TraceString()
190 void RecvOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()
219 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_CPU), RecvOp);
220 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_DEFAULT), RecvOp);
221 REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_TPU_SYSTEM), RecvOp);
222 REGISTER_KERNEL_BUILDER(Name("_HostRecv").Device(DEVICE_TPU_SYSTEM), RecvOp);
225 REGISTER_KERNEL_BUILDER(Name("Recv").Device(DEVICE_CPU), RecvOp);
226 REGISTER_KERNEL_BUILDER(Name("Recv").Device(DEVICE_DEFAULT), RecvOp);
229 Name("_HostRecv").Device(DEVICE_DEFAULT).HostMemory("tensor"), RecvOp);
[all …]
Dsendrecv_ops.h39 class RecvOp : public AsyncOpKernel {
41 explicit RecvOp(OpKernelConstruction* ctx);
51 TF_DISALLOW_COPY_AND_ASSIGN(RecvOp);
/external/tensorflow/tensorflow/core/framework/
Drendezvous.h76 friend class RecvOp; member
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc327 } else if (isa<mlir::mhlo::RecvOp>(op)) { in MhloToHloOpcode()
Dmlir_hlo_to_hlo.cc837 LogicalResult ExportXlaOp(RecvOp op, OpLoweringContext ctx) { in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc286 builder.create<RecvOp>(loc, recv_result_type, token, channel_handle, in CreateRecvOp()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc1700 static LogicalResult Verify(RecvOp op) { in Verify()