Searched refs:StackCloseOp (Results 1 – 5 of 5) sorted by relevance
196 REGISTER_KERNEL_BUILDER(Name("StackClose").Device(DEVICE_CPU), StackCloseOp);198 Name("StackClose").Device(DEVICE_GPU).HostMemory("handle"), StackCloseOp);199 REGISTER_KERNEL_BUILDER(Name("StackCloseV2").Device(DEVICE_CPU), StackCloseOp);201 Name("StackCloseV2").Device(DEVICE_GPU).HostMemory("handle"), StackCloseOp);204 Name("StackClose").Device(DEVICE_SYCL).HostMemory("handle"), StackCloseOp);207 StackCloseOp);
67 class StackCloseOp : public OpKernel {69 explicit StackCloseOp(OpKernelConstruction* context);
328 StackCloseOp::StackCloseOp(OpKernelConstruction* context) : OpKernel(context) {} in StackCloseOp() function in tensorflow::StackCloseOp330 void StackCloseOp::Compute(OpKernelContext* ctx) { in Compute()337 bool StackCloseOp::IsExpensive() { return false; } in IsExpensive()
228 class StackCloseOp : public XlaOpKernel { class230 explicit StackCloseOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in StackCloseOp() function in tensorflow::__anon86d884060111::StackCloseOp237 TF_DISALLOW_COPY_AND_ASSIGN(StackCloseOp);240 REGISTER_XLA_OP(Name("StackCloseV2").CompilationOnly(), StackCloseOp);
300 Name("StackCloseV2").Device(DEVICE).HostMemory("handle"), StackCloseOp);