1 write to _host
Microsoft.ML.TensorFlow (1)
TensorflowTransform.cs (1)
1003_host = Contracts.CheckRef(env, nameof(env)).Register(nameof(TensorFlowEstimator));
8 references to _host
Microsoft.ML.TensorFlow (8)
TensorflowTransform.cs (8)
1009var inputTuple = TensorFlowTransformer.GetInputInfo(_host, tensorFlowModel.Session, _options.InputColumns); 1011var outputTuple = TensorFlowTransformer.GetOutputInfo(_host, tensorFlowModel.Session, _options.OutputColumns, _options.TreatOutputAsBatched); 1032_host.CheckValue(inputSchema, nameof(inputSchema)); 1039throw _host.ExceptSchemaMismatch(nameof(inputSchema), "input", input); 1042throw _host.ExceptSchemaMismatch(nameof(inputSchema), "input", input, expectedType.ToString(), col.ItemType.ToString()); 1058_host.CheckValue(input, nameof(input)); 1061_transformer = new TensorFlowTransformer(_host, _tensorFlowModel.Session, _options.OutputColumns, _options.InputColumns, 1062IsSavedModel(_host, _options.ModelLocation) ? _options.ModelLocation : null, false, _options.AddBatchDimensionInputs, treatOutputAsBatched: _options.TreatOutputAsBatched);