OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:EnterOp
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
control_flow_ops.h
65
class
EnterOp
: public OpKernel {
67
explicit
EnterOp
(OpKernelConstruction* context) : OpKernel(context) {}
70
~
EnterOp
() override {}
72
TF_DISALLOW_COPY_AND_ASSIGN(
EnterOp
);
control_flow_ops.cc
336
void
EnterOp
::Compute(OpKernelContext* context) {
344
REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_CPU),
EnterOp
);
345
REGISTER_KERNEL_BUILDER(Name("RefEnter").Device(DEVICE_CPU),
EnterOp
);
349
Name("Enter").Device(DEVICE_GPU).TypeConstraint<type>("T"),
EnterOp
)
352
Name("RefEnter").Device(DEVICE_GPU).TypeConstraint<type>("T"),
EnterOp
)
365
Name("Enter").Device(DEVICE_SYCL).TypeConstraint<type>("T"),
EnterOp
)
371
Name("RefEnter").Device(DEVICE_SYCL).TypeConstraint<type>("T"),
EnterOp
)
383
EnterOp
)
391
EnterOp
)
412
EnterOp
)
[
all
...]
Completed in 178 milliseconds