HomeSort by relevance Sort by last modified time
    Searched refs:StackCloseOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
stack_ops.cc 235 class StackCloseOp : public XlaOpKernel {
237 explicit StackCloseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
244 TF_DISALLOW_COPY_AND_ASSIGN(StackCloseOp);
247 REGISTER_XLA_OP(Name("StackCloseV2"), StackCloseOp);
  /external/tensorflow/tensorflow/core/kernels/
stack_ops.cc 503 class StackCloseOp : public OpKernel {
505 explicit StackCloseOp(OpKernelConstruction* context) : OpKernel(context) {}
517 REGISTER_KERNEL_BUILDER(Name("StackClose").Device(DEVICE_CPU), StackCloseOp);
519 Name("StackClose").Device(DEVICE_GPU).HostMemory("handle"), StackCloseOp);
520 REGISTER_KERNEL_BUILDER(Name("StackCloseV2").Device(DEVICE_CPU), StackCloseOp);
522 Name("StackCloseV2").Device(DEVICE_GPU).HostMemory("handle"), StackCloseOp);
525 Name("StackClose").Device(DEVICE_SYCL).HostMemory("handle"), StackCloseOp);
528 StackCloseOp);

Completed in 84 milliseconds