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()77 TF_DISALLOW_COPY_AND_ASSIGN(RemoteCallOp);
296 RemoteCallOp::RemoteCallOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { in RemoteCallOp() function in tensorflow::RemoteCallOp303 void RemoteCallOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()397 Name("RemoteCall").Device(DEVICE_CPU).HostMemory("target"), RemoteCallOp);399 Name("RemoteCall").Device(DEVICE_GPU).HostMemory("target"), RemoteCallOp);402 Name("RemoteCall").Device(DEVICE_SYCL).HostMemory("target"), RemoteCallOp);
231 Name("RemoteCall").Device(DEVICE).HostMemory("target"), RemoteCallOp); \