Function get_device_option

caffe2/python/onnx/backend.py:45–48 · view source on GitHub ↗

(device)

Source from the content-addressed store, hash-verified

43	return s
44
45	def get_device_option(device):
46	m = {DeviceType.CPU: caffe2_pb2.CPU,
47	DeviceType.CUDA: workspace.GpuDeviceType}
48	return core.DeviceOption(m[device.type], device.device_id)
49
50
51	class OnnxAttributes(dict):

run_nodeMethod · 0.85

prepareMethod · 0.85

_onnx_model_to_caffe2_netMethod · 0.85

no outgoing calls

no test coverage detected

searching dependent graphs…