Searched refs:StackPushOp (Results 1 – 3 of 3) sorted by relevance
41 class StackPushOp : public AsyncOpKernel {43 StackPushOp(OpKernelConstruction* context, bool allow_swapping);54 class TemplatedStackPushOp : public StackPushOp {57 : StackPushOp(context, allow_swapping) {} in TemplatedStackPushOp()
213 StackPushOp::StackPushOp(OpKernelConstruction* context, bool allow_swapping) in StackPushOp() function in tensorflow::StackPushOp220 void StackPushOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()282 bool StackPushOp::IsExpensive() { return false; } in IsExpensive()
127 class StackPushOp : public XlaOpKernel { class129 explicit StackPushOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in StackPushOp() function in tensorflow::__anon86d884060111::StackPushOp169 TF_DISALLOW_COPY_AND_ASSIGN(StackPushOp);172 REGISTER_XLA_OP(Name("StackPushV2").CompilationOnly(), StackPushOp);