5 writes to GpuDeviceId
Microsoft.ML.OnnxTransformer (5)
OnnxTransform.cs (5)
224options.GpuDeviceId = localEnvironment.GpuDeviceId; 319GpuDeviceId = gpuDeviceId, 345GpuDeviceId = gpuDeviceId, 375GpuDeviceId = gpuDeviceId, 407GpuDeviceId = gpuDeviceId,
3 references to GpuDeviceId
Microsoft.ML.OnnxTransformer (3)
OnnxTransform.cs (3)
265Model = new OnnxModel(env, options.ModelFile, options.GpuDeviceId, options.FallbackToCpu, ownModelFile: false, shapeDictionary: shapeDictionary, options.RecursionLimit, 273Model = OnnxModel.CreateFromStream(options.ModelBytes, env, options.GpuDeviceId, options.FallbackToCpu, shapeDictionary: shapeDictionary, options.RecursionLimit); 280Model = OnnxModel.CreateFromBytes(modelBytes, env, options.GpuDeviceId, options.FallbackToCpu, shapeDictionary: shapeDictionary, options.RecursionLimit);