Home
last modified time | relevance | path

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

/external/pytorch/torch/autograd/
Dprofiler_legacy.py164 def key_averages(self, group_by_input_shape=False, group_by_stack_n=0): argument
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
Dprofiler.py480 def key_averages(self, group_by_input_shape=False, group_by_stack_n=0): argument