Searched refs:RemoteCallOp (Results 1 – 3 of 3) sorted by relevance
59 class RemoteCallOp : public AsyncOpKernel {61 explicit RemoteCallOp(OpKernelConstruction* ctx);63 ~RemoteCallOp() override {} in ~RemoteCallOp()79 TF_DISALLOW_COPY_AND_ASSIGN(RemoteCallOp);
264 RemoteCallOp::RemoteCallOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in RemoteCallOp() function in tensorflow::RemoteCallOp271 void RemoteCallOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()380 string RemoteCallOp::TraceString(const OpKernelContext& ctx, in TraceString()395 Name("RemoteCall").Device(DEVICE_CPU).HostMemory("target"), RemoteCallOp);397 Name("RemoteCall").Device(DEVICE_GPU).HostMemory("target"), RemoteCallOp);
176 Name("RemoteCall").Device(DEVICE).HostMemory("target"), RemoteCallOp); \