Searched refs:get_kernel_args (Results 1 – 1 of 1) sorted by relevance
1258 def get_kernel_args(self, node_schedule, numel, reduction_numel): member in SIMDScheduling1305 ) = self.get_kernel_args(node_schedule, numel, reduction_numel)1533 ) = self.get_kernel_args(node_schedule, numel, rnumel)1790 reduction_hint_val, mutations, index_dtype = self.get_kernel_args(