Home
last modified time | relevance | path

Searched refs:group_by_input_shapes (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/autograd/
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
325 if group_by_input_shapes:
332 stats[get_key(evt, group_by_input_shapes, group_by_stack_n)].add(evt)
342 if not group_by_input_shapes: