Searched defs:tensor_list_args (Results 1 – 1 of 1) sorted by relevance
166 FuncOp func, const llvm::SmallSet<int, 4> &tensor_list_args) { in GetResizedTensorListIndexes()1102 ArrayRef<Value> operands, const llvm::SmallSet<int, 4> &tensor_list_args, in UpdateFunctionTypesForWhileOp()1148 const llvm::SmallSet<int, 4> &tensor_list_args, in UpdateFunctionTypesForIfOp()1265 auto tensor_list_args = GetTensorListArgumentsIndex(op.else_function()); in matchAndRewrite() local1302 auto tensor_list_args = GetTensorListArgumentsIndex(op.body_function()); in matchAndRewrite() local