OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:RecvOp
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
sendrecv_ops.h
37
class
RecvOp
: public AsyncOpKernel {
39
explicit
RecvOp
(OpKernelConstruction* ctx);
47
TF_DISALLOW_COPY_AND_ASSIGN(
RecvOp
);
sendrecv_ops.cc
123
RecvOp
::
RecvOp
(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) {
171
void
RecvOp
::ComputeAsync(OpKernelContext* ctx, DoneCallback done) {
196
REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_CPU),
RecvOp
);
197
REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_GPU),
RecvOp
);
200
REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE_SYCL),
RecvOp
);
203
REGISTER_KERNEL_BUILDER(Name("_HostRecv").Device(DEVICE_CPU),
RecvOp
);
205
Name("_HostRecv").Device(DEVICE_GPU).HostMemory("tensor"),
RecvOp
);
209
Name("_HostRecv").Device(DEVICE_SYCL).HostMemory("tensor"),
RecvOp
);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
sendrecv_ops.cc
53
class
RecvOp
: public XlaOpKernel {
55
explicit
RecvOp
(OpKernelConstruction* ctx);
62
TF_DISALLOW_COPY_AND_ASSIGN(
RecvOp
);
65
RecvOp
::
RecvOp
(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {
75
void
RecvOp
::Compile(XlaOpKernelContext* ctx) {
82
REGISTER_XLA_OP(Name("_XLARecv"),
RecvOp
);
/external/tensorflow/tensorflow/compiler/jit/
xla_device_ops.h
52
REGISTER_KERNEL_BUILDER(Name("_Recv").Device(DEVICE),
RecvOp
); \
56
Name("_HostRecv").Device(DEVICE).HostMemory("tensor"),
RecvOp
); \
/external/tensorflow/tensorflow/core/framework/
rendezvous.h
79
friend class
RecvOp
;
Completed in 206 milliseconds