Searched refs:node_cpp (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/tensorboard/ |
| H A D | _pytorch_graph.py | 67 def __init__(self, node_cpp, valid_methods): argument
68 super().__init__(node_cpp)
74 list_of_node = list(getattr(node_cpp, m)())
88 setattr(self, m, getattr(node_cpp, m)())
92 def __init__(self, node_cpp, input_or_output=None): argument
93 super().__init__(node_cpp, methods_IO)
95 tensor_size = node_cpp.type().sizes()
112 def __init__(self, node_cpp): argument
113 super().__init__(node_cpp, methods_OP)
117 {k: _node_get(node_cpp, k) for k in node_cpp.attributeNames()}
[all …]
|