Searched defs:ExitOp (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcontrol_flow_ops.h78 class ExitOp : public OpKernel { class in namespace:tensorflow
80 explicit ExitOp(OpKernelConstruction* context) : OpKernel(context) {} function in class:tensorflow::ExitOp
83 ~ExitOp() override {}
85 TF_DISALLOW_COPY_AND_ASSIGN(ExitOp);
H A Dcontrol_flow_ops.cc431 void ExitOp::Compute(OpKernelContext* context) {
439 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_CPU), ExitOp); member in namespace:tensorflow
440 REGISTER_KERNEL_BUILDER(Name("RefExit").Device(DEVICE_CPU), ExitOp); member in namespace:tensorflow
444 Name("Exit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp);
447 Name("RefExit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp);
460 Name("Exit").Device(DEVICE_SYCL).TypeConstraint<type>("T"), ExitOp); \
462 Name("RefExit").Device(DEVICE_SYCL).TypeConstraint<type>("T"), ExitOp);
475 ExitOp); \
481 ExitOp)
497 ExitOp); \
[all...]

Completed in 86 milliseconds