Searched refs:tensor_list_args (Results 1 – 1 of 1) sorted by relevance
837 llvm::SmallSet<int, 4> tensor_list_args) { in UpdateFunctionTypes() argument853 if (tensor_list_args.count(i)) { in UpdateFunctionTypes()870 if (func_index != 1 && tensor_list_args.count(i)) { in UpdateFunctionTypes()900 auto tensor_list_args = GetTensorListArgumentsFromWhileOp(op); in matchAndRewrite() local907 if (tensor_list_args.count(i)) { in matchAndRewrite()920 (void)UpdateFunctionTypes(converted, tensor_list_args); in matchAndRewrite()