Searched refs:StackPopOp (Results 1 – 5 of 5) sorted by relevance
131 REGISTER_KERNEL_BUILDER(Name("StackPop").Device(DEVICE_CPU), StackPopOp);132 REGISTER_KERNEL_BUILDER(Name("StackPopV2").Device(DEVICE_CPU), StackPopOp);139 StackPopOp); \144 StackPopOp);158 StackPopOp); \164 StackPopOp);177 StackPopOp)187 StackPopOp)
60 class StackPopOp : public AsyncOpKernel {62 explicit StackPopOp(OpKernelConstruction* context);
286 StackPopOp::StackPopOp(OpKernelConstruction* context) in StackPopOp() function in tensorflow::StackPopOp289 void StackPopOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()324 bool StackPopOp::IsExpensive() { return false; } in IsExpensive()
174 class StackPopOp : public XlaOpKernel { class176 explicit StackPopOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in StackPopOp() function in tensorflow::__anon86d884060111::StackPopOp223 TF_DISALLOW_COPY_AND_ASSIGN(StackPopOp);226 REGISTER_XLA_OP(Name("StackPopV2").CompilationOnly(), StackPopOp);
298 StackPopOp); \