Searched refs:arg_number (Results 1 – 6 of 6) sorted by relevance
102 def match_arg_dtype(arg_number): argument103 arg = args[arg_number]104 if not arg_is_tensor[arg_number]:107 'was %s instead' % (arg_number, type(arg)))112 return_dtypes = match_arg_dtype(return_dtypes.arg_number)115 match_arg_dtype(a.arg_number) if isinstance(a, MatchDType) else a
653 unsigned arg_number = block_arg.getArgNumber(); in GetOperandsForBlockArgument() local660 if (arg_number < num_replicated_args) in GetOperandsForBlockArgument()661 return operands.slice(arg_number * num_replicas, num_replicas); in GetOperandsForBlockArgument()664 arg_number -= num_replicated_args; in GetOperandsForBlockArgument()665 return operands.slice(arg_number, 1); in GetOperandsForBlockArgument()
187 unsigned arg_number = arg.getArgNumber(); in LiftVariables() local189 func, arg_number, symbol_table); in LiftVariables()
122 new_type = IntegerType.get_signless(8 * (arg.arg_number + 1))
2204 def arg_number(self, num): member in DictCompiler2220 data.append(self.arg_number(num))
1230 def arg_number(self, name): member in DictDecompiler