MCPcopy
hub / github.com/apache/tvm / cpu

Method cpu

python/tvm/rpc/client.py:219–221  ·  view source on GitHub ↗

Construct CPU device.

(self, dev_id=0)

Source from the content-addressed store, hash-verified

217 return self._remote_funcs["download_linked_module"](path)
218
219 def cpu(self, dev_id=0):
220 """Construct CPU device."""
221 return self.device(DLDeviceType.kDLCPU, dev_id)
222
223 def cuda(self, dev_id=0):
224 """Construct CUDA GPU device."""

Callers 14

_convert_single_pytorch_to_tvmMethod · 0.45
_relax_backendFunction · 0.45
from_fxMethod · 0.45
_convert_torch_tensor_to_relaxMethod · 0.45
_convert_pytorch_tensor_to_tvmMethod · 0.45
coreml_compilerFunction · 0.45
__call__Method · 0.45
benchmarkFunction · 0.45
train_stepMethod · 0.45
predict_stepMethod · 0.45
get_runtime_deviceFunction · 0.45
buildFunction · 0.45

Calls 1

deviceMethod · 0.95

Tested by

no test coverage detected