22 references to _fallbackToCpu
Microsoft.ML.OnnxTransformerTest (22)
OnnxTransformTests.cs (22)
121
ML.FallbackToCpu =
_fallbackToCpu
;
152
FallbackToCpu =
_fallbackToCpu
,
158
ML.Transforms.ApplyOnnxModel(options.OutputColumns, options.InputColumns, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
257
.Append(ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", fileStream, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
));
308
.Append(ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
));
369
var pipeline = ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
408
var pipeline = ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
417
Assert.Throws<InvalidOperationException>(() => ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
));
436
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outa", "outb" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
476
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outb", "outa" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
502
pipeline = ML.Transforms.ApplyOnnxModel(new[] { "outb" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
536
var pipeline = ML.Transforms.ApplyOnnxModel(modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
562
var pipeline = ML.Transforms.ApplyOnnxModel(modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
644
.Append(ML.Transforms.ApplyOnnxModel("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
)); // Map column "data_0" to column "softmaxout_1"
694
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
747
var pipeline = ML.Transforms.ApplyOnnxModel(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
911
modelFile, shapeDictionary, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
918
modelFile, shapeDictionary, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
923
pipeline[2] = ML.Transforms.ApplyOnnxModel(modelFile, shapeDictionary, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
973
modelFile, shapeDictionary, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
1073
nameof(InputWithCustomShape.input), modelFile, shapeDictionary, gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
);
1138
gpuDeviceId: _gpuDeviceId, fallbackToCpu:
_fallbackToCpu
, shapeDictionary: null, recursionLimit: 50));