Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_device_analysis.py37 def node_output_device(graph): member in TestDeviceAnalysis
68 actual_device = self.node_output_device(graph)
102 actual_device = self.node_output_device(graph)
203 actual_device = self.node_output_device(graph)