2 instantiations of TensorFlowSessionWrapper
Microsoft.ML.TensorFlow (1)
TensorflowUtils.cs (1)
397new TensorFlowSessionWrapper(GetSession(env, modelPath, metaGraph), modelPath);
Microsoft.ML.Vision (1)
ImageClassificationTrainer.cs (1)
1316return new TensorFlowSessionWrapper(GetSession(env, modelFilePath, true), modelFilePath);
7 references to TensorFlowSessionWrapper
Microsoft.ML.TensorFlow (2)
TensorFlowSessionWrapper.cs (1)
18/// Instantiates <see cref="TensorFlowSessionWrapper"/>.
TensorflowUtils.cs (1)
396internal static TensorFlowSessionWrapper LoadDnnModel(IHostEnvironment env, string modelPath, bool metaGraph = false) =>
Microsoft.ML.Vision (5)
DnnRetrainTransform.cs (4)
171internal DnnRetrainTransformer(IHostEnvironment env, DnnRetrainEstimator.Options options, ML.TensorFlow.TensorFlowSessionWrapper tensorFlowModel, IDataView input, IDataView validationSet = null) 1223private readonly ML.TensorFlow.TensorFlowSessionWrapper _tensorFlowModel; 1228internal DnnRetrainEstimator(IHostEnvironment env, Options options, ML.TensorFlow.TensorFlowSessionWrapper tensorFlowModel) 1238private static Options CreateArguments(ML.TensorFlow.TensorFlowSessionWrapper tensorFlowModel, string[] outputColumnNames, string[] inputColumnName, bool addBatchDimensionInput)
ImageClassificationTrainer.cs (1)
1310private TensorFlowSessionWrapper LoadTensorFlowSessionFromMetaGraph(IHostEnvironment env, Architecture arch)