Searched defs:tensor_list_args (Results 1 – 1 of 1) sorted by relevance
158 func::FuncOp func, const llvm::SmallSet<int, 4> &tensor_list_args) { in GetResizedTensorListIndexes()1210 const llvm::SmallSet<int, 4> &tensor_list_args, in UpdateFunctionTypesForWhileOp()1255 const llvm::SmallSet<int, 4> &tensor_list_args, in UpdateFunctionTypesForIfOp()1372 auto tensor_list_args = GetTensorListArgumentsIndex(op.else_function()); in matchAndRewrite() local1409 auto tensor_list_args = GetTensorListArgumentsIndex(op.body_function()); in matchAndRewrite() local