Searched refs:_get_onnx_devices (Results 1 – 1 of 1) sorted by relevance
317 def _get_onnx_devices( function1027 input_devices = _get_onnx_devices(args)1032 output_devices = _get_onnx_devices(prim_outputs)1034 output_devices = _get_onnx_devices((prim_outputs,))