Home
last modified time | relevance | path

Searched defs:group_by_stack_n (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Dprofiler_util.py299 def key_averages(self, group_by_input_shapes=False, group_by_stack_n=0): argument
317 def get_key(event, group_by_input_shapes, group_by_stack_n) -> Tuple[str, ...]: argument
H A Dprofiler_legacy.py164 def key_averages(self, group_by_input_shape=False, group_by_stack_n=0): argument
H A Dprofiler.py480 def key_averages(self, group_by_input_shape=False, group_by_stack_n=0): argument