Home
last modified time | relevance | path

Searched refs:SendOp (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsendrecv_ops.cc29 class SendOp : public XlaOpKernel { class
31 explicit SendOp(OpKernelConstruction* ctx);
37 TF_DISALLOW_COPY_AND_ASSIGN(SendOp);
40 SendOp::SendOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SendOp() function in tensorflow::__anon233ac8c10111::SendOp
44 void SendOp::Compile(XlaOpKernelContext* ctx) { in Compile()
51 REGISTER_XLA_OP(Name("XlaSend"), SendOp);
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc58 SendOp::SendOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in SendOp() function in tensorflow::SendOp
80 void SendOp::Compute(OpKernelContext* ctx) { in Compute()
115 string SendOp::TraceString(const OpKernelContext& ctx, bool verbose) const { in TraceString()
125 REGISTER_KERNEL_BUILDER(Name("_Send").Device(DEVICE_CPU), SendOp);
126 REGISTER_KERNEL_BUILDER(Name("_Send").Device(DEVICE_DEFAULT), SendOp);
127 REGISTER_KERNEL_BUILDER(Name("_Send").Device(DEVICE_TPU_SYSTEM), SendOp);
128 REGISTER_KERNEL_BUILDER(Name("_HostSend").Device(DEVICE_TPU_SYSTEM), SendOp);
131 REGISTER_KERNEL_BUILDER(Name("Send").Device(DEVICE_CPU), SendOp);
132 REGISTER_KERNEL_BUILDER(Name("Send").Device(DEVICE_DEFAULT), SendOp);
135 Name("_HostSend").Device(DEVICE_DEFAULT).HostMemory("tensor"), SendOp);
[all …]
Dsendrecv_ops.h24 class SendOp : public OpKernel {
26 explicit SendOp(OpKernelConstruction* ctx);
36 TF_DISALLOW_COPY_AND_ASSIGN(SendOp);
/external/tensorflow/tensorflow/core/framework/
Drendezvous.h75 friend class SendOp; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcross_host_transfer.cc132 builder.create<tf_device::SendOp>(op->getLoc(), arg, key, dst_host); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc325 } else if (isa<mlir::mhlo::SendOp>(op)) { in MhloToHloOpcode()
Dmlir_hlo_to_hlo.cc983 LogicalResult ExportXlaOp(SendOp op, OpLoweringContext ctx) { in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc261 auto send = builder.create<SendOp>( in CreateSendOp()