Searched refs:TensorListPushBackOp (Results 1 – 5 of 5) sorted by relevance
631 class TensorListPushBackOp : public XlaOpKernel { class633 explicit TensorListPushBackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TensorListPushBackOp() function in tensorflow::__anond80ffd040111::TensorListPushBackOp653 TF_DISALLOW_COPY_AND_ASSIGN(TensorListPushBackOp);657 TensorListPushBackOp);
516 : public OpConversionPattern<TF::TensorListPushBackOp> {520 TF::TensorListPushBackOp op, ArrayRef<Value> operands, in matchAndRewrite()820 llvm::isa<TF::TensorListPushBackOp>(op) || in GetTensorListArgumentsFromWhileOp()992 TF::TensorListPushBackOp, TF::TensorListReserveOp, in runOnOperation()
134 GET_OPERATION_NAME(TF::TensorListPushBackOp), in AddRewrittenCompositeOps()
636 TF::TensorListPushBackOp push, in HandleTensorListPushBackOp()836 } else if (auto push = llvm::dyn_cast<TF::TensorListPushBackOp>(&op)) { in DecomposeTensorListOpsInternal()
193 if (auto push = llvm::dyn_cast<TensorListPushBackOp>(use.getOwner())) { in CanInferTensorListElementType()