HomeSort by relevance Sort by last modified time
    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