19 references to _fallbackToCpu
Microsoft.ML.OnnxTransformerTest (19)
OnnxTransformTests.cs (19)
120
ML.FallbackToCpu =
_fallbackToCpu
;
151
FallbackToCpu =
_fallbackToCpu
,
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));