1 write to Session
Microsoft.ML.TensorFlow (1)
TensorFlowModel.cs (1)
34Session = session;
11 references to Session
Microsoft.ML.TensorFlow (11)
TensorFlowModel.cs (3)
46return TensorFlowUtils.GetModelSchema(_env, Session.graph, TreatOutputAsBatched); 55return TensorFlowUtils.GetModelSchema(_env, Session.graph, TreatOutputAsBatched, "Placeholder"); 100Session.Dispose();
TensorflowTransform.cs (7)
105: this(env, tfModelInfo.Session, new[] { outputColumnName }, new[] { inputColumnName ?? outputColumnName }, IsSavedModel(env, tfModelInfo.ModelPath) ? tfModelInfo.ModelPath : null, false, addBatchDimensionInput, treatOutputAsBatched: treatOutputAsBatched) 123: this(env, tfModelInfo.Session, outputColumnNames, inputColumnNames, IsSavedModel(env, tfModelInfo.ModelPath) ? tfModelInfo.ModelPath : null, false, addBatchDimensionInput, treatOutputAsBatched: treatOutputAsBatched) 207: this(env, tensorFlowModel.Session, options.OutputColumns, options.InputColumns, 1008tensorFlowModel.Session.graph.as_default(); 1009var inputTuple = TensorFlowTransformer.GetInputInfo(_host, tensorFlowModel.Session, _options.InputColumns); 1011var outputTuple = TensorFlowTransformer.GetOutputInfo(_host, tensorFlowModel.Session, _options.OutputColumns, _options.TreatOutputAsBatched); 1061_transformer = new TensorFlowTransformer(_host, _tensorFlowModel.Session, _options.OutputColumns, _options.InputColumns,
TensorflowUtils.cs (1)
132return GetModelSchema(env, model.Session.graph, treatOutputAsBatched);