19 references to _gpuDeviceId
Microsoft.ML.OnnxTransformerTest (19)
OnnxTransformTests.cs (19)
119
ML.GpuDeviceId =
_gpuDeviceId
;
150
GpuDeviceId =
_gpuDeviceId
,
157
ML.Transforms.ApplyOnnxModel(options.OutputColumns, options.InputColumns, modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
254
.Append(ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu));
315
var pipeline = ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
348
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outa", "outb" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
388
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outb", "outa" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
414
pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outb" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
448
var pipeline = ML.Transforms.ApplyOnnxModel(modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
474
var pipeline = ML.Transforms.ApplyOnnxModel(modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
556
.Append(ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu)); // Map column "data_0" to column "softmaxout_1"
606
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
659
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
823
modelFile, shapeDictionary, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
830
modelFile, shapeDictionary, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
835
pipeline[2] = ML.Transforms.ApplyOnnxModel(modelFile, shapeDictionary, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
885
modelFile, shapeDictionary, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
985
nameof(InputWithCustomShape.input), modelFile, shapeDictionary, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
1050
gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu, shapeDictionary: null, recursionLimit: 50));