Searched defs:_get_tensor (Results 1 – 5 of 5) sorted by relevance
168 def _get_tensor(name): function
187 def _get_tensor(node): function
104 def _get_tensor(self, is_finite): member in DynamicLossScaleTest
97 def _get_tensor(node, index): function
58 def _get_tensor(node: torch.fx.Node) -> torch.Tensor: function