Searched refs:EnterOp (Results 1 – 3 of 3) sorted by relevance
65 class EnterOp : public OpKernel {67 explicit EnterOp(OpKernelConstruction* context) : OpKernel(context) {} in EnterOp() function70 ~EnterOp() override {} in ~EnterOp()72 TF_DISALLOW_COPY_AND_ASSIGN(EnterOp);
347 void EnterOp::Compute(OpKernelContext* context) { in Compute()355 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_CPU), EnterOp);356 REGISTER_KERNEL_BUILDER(Name("RefEnter").Device(DEVICE_CPU), EnterOp);360 Name("Enter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)363 Name("RefEnter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)377 Name("Enter").Device(DEVICE_SYCL).TypeConstraint<type>("T"), EnterOp)383 Name("RefEnter").Device(DEVICE_SYCL).TypeConstraint<type>("T"), EnterOp)395 EnterOp)403 EnterOp)424 EnterOp)[all …]
182 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE), EnterOp); \