Searched refs:TensorListStackOp (Results 1 – 6 of 6) sorted by relevance
430 class TensorListStackOp : public XlaOpKernel { class432 explicit TensorListStackOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TensorListStackOp() function in tensorflow::__anond80ffd040111::TensorListStackOp455 TF_DISALLOW_COPY_AND_ASSIGN(TensorListStackOp);458 REGISTER_XLA_OP(Name("TensorListStack"), TensorListStackOp);
669 auto stacked_extended_part = rewriter->create<TF::TensorListStackOp>( in CreateCondTrueBranch()746 : public OpConversionPattern<TF::TensorListStackOp> {750 TF::TensorListStackOp op, ArrayRef<Value> operands, in matchAndRewrite()823 llvm::isa<TF::TensorListStackOp>(op) || in GetTensorListArgumentsFromWhileOp()993 TF::TensorListSetItemOp, TF::TensorListStackOp, in runOnOperation()
142 GET_OPERATION_NAME(TF::TensorListStackOp), in AddRewrittenCompositeOps()
856 } else if (auto stack = llvm::dyn_cast<TF::TensorListStackOp>(&op)) { in DecomposeTensorListOpsInternal()
270 if (isa<TensorListStackOp, TensorListGatherOp, TensorListConcatV2Op, in CanInferTensorListElementType()
2206 static LogicalResult Verify(TensorListStackOp op) { in Verify()