Searched defs:numel_input (Results 1 – 2 of 2) sorted by relevance
66 output_tensors: Tuple, numel_input, dtype=None, device=None argument
397 auto numel_input = at::symint::numel<T>(input); in needs_64bit_indexing_no_split() local