Searched defs:rnumel (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | triton_helpers.py | 407 rnumel, argument
476 rnumel, argument
509 rnumel, # number of elements argument
|
| H A D | triton_heuristics.py | 1728 def split_scan_grid(xnumel, rnumel): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | simd.py | 1065 def generate_node_schedule(self, nodes, numel, rnumel): argument
1235 def has_non_contiguous_pw_in_reduction_kernel(self, node_schedule, numel, rnumel): argument
|
| H A D | triton.py | 2752 def _get_persistent_RBLOCK(self, rnumel): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 1856 const c10::SymFloat rnumel(_safe_size(self.sym_sizes(), dim)); in var_backward() local
|