MCPcopy Create free account
hub / github.com/pytorch/pytorch / generateDevices

Method generateDevices

caffe2/python/docs/generator.py:190–198  ·  view source on GitHub ↗
(self, formatter)

Source from the content-addressed store, hash-verified

188 pass
189
190 def generateDevices(self, formatter):
191 formatter.addHeader("Devices", 3)
192 devices = [
193 self.getInfo(formatter,
194 'CPU', OpSchema.get_cpu_impl(self.name)),
195 self.getInfo(formatter,
196 'GPU', OpSchema.get_cuda_impl(self.name)),
197 ]
198 formatter.addList([i for i in devices if i])
199
200 def generateEngines(self, formatter):
201 if not len(self.engines):

Callers 1

generateSchemaMethod · 0.95

Calls 3

getInfoMethod · 0.95
addHeaderMethod · 0.45
addListMethod · 0.45

Tested by

no test coverage detected