Searched refs:ExitOp (Results 1 – 3 of 3) sorted by relevance
78 class ExitOp : public OpKernel {80 explicit ExitOp(OpKernelConstruction* context) : OpKernel(context) {} in ExitOp() function83 ~ExitOp() override {} in ~ExitOp()85 TF_DISALLOW_COPY_AND_ASSIGN(ExitOp);
443 void ExitOp::Compute(OpKernelContext* context) { in Compute()451 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_CPU), ExitOp);452 REGISTER_KERNEL_BUILDER(Name("RefExit").Device(DEVICE_CPU), ExitOp);456 Name("Exit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp);459 Name("RefExit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp);473 Name("Exit").Device(DEVICE_SYCL).TypeConstraint<type>("T"), ExitOp); \475 Name("RefExit").Device(DEVICE_SYCL).TypeConstraint<type>("T"), ExitOp);488 ExitOp); \494 ExitOp)510 ExitOp); \[all …]
183 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE), ExitOp); \