41 references to TensorFlowUtils
Microsoft.ML.TensorFlow (15)
TensorflowCatalog.cs (2)
37=> TensorFlowUtils.LoadTensorFlowModel(CatalogUtils.GetEnvironment(catalog), modelLocation); 63=> TensorFlowUtils.LoadTensorFlowModel(CatalogUtils.GetEnvironment(catalog), modelLocation, treatOutputAsBatched);
TensorFlowModel.cs (2)
46return TensorFlowUtils.GetModelSchema(_env, Session.graph, TreatOutputAsBatched); 55return TensorFlowUtils.GetModelSchema(_env, Session.graph, TreatOutputAsBatched, "Placeholder");
TensorflowTransform.cs (11)
21using static Microsoft.ML.TensorFlow.TensorFlowUtils; 95/// It is useful in a situation where user has already loaded TensorFlow model using <see cref="TensorFlowUtils.LoadTensorFlowModel(IHostEnvironment, string, bool)"/> for inspecting model schema. 98/// <param name="tfModelInfo"> <see cref="TensorFlowModel"/> object created with <see cref="TensorFlowUtils.LoadTensorFlowModel(IHostEnvironment, string, bool)"/>.</param> 113/// It is useful in a situation where user has already loaded TensorFlow model using <see cref="TensorFlowUtils.LoadTensorFlowModel(IHostEnvironment, string, bool)"/> for inspecting model schema. 116/// <param name="tfModelInfo"> <see cref="TensorFlowModel"/> object created with <see cref="TensorFlowUtils.LoadTensorFlowModel(IHostEnvironment, string, bool)"/>.</param> 202: this(env, options, TensorFlowUtils.LoadTensorFlowModel(env, options.ModelLocation), input) 845var tensor = TensorFlowUtils.CastDataAndReturnAsTensor(scalar); 905var tensor = TensorFlowUtils.CastDataAndReturnAsTensor(_denseData, _tfShape); 919var tensor = TensorFlowUtils.CastDataAndReturnAsTensor(_denseData, _tfShape); 987: this(env, outputColumnNames, inputColumnNames, TensorFlowUtils.LoadTensorFlowModel(env, modelLocation), addBatchDimensionInput) 997: this(env, options, TensorFlowUtils.LoadTensorFlowModel(env, options.ModelLocation, options.TreatOutputAsBatched))
Microsoft.ML.TensorFlow.Tests (19)
TensorFlowEstimatorTests.cs (1)
241using var tensorFlowModel = TensorFlowUtils.LoadTensorFlowModel(mlContext, modelLocation);
TensorflowTests.cs (18)
580var schema = TensorFlowUtils.GetModelSchema(_mlContext, modelLocation); 587var metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 591schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 593metadataType = schema[col].Annotations.Schema.GetColumnOrNull(TensorFlowUtils.TensorflowUpstreamOperatorsKind)?.Type; 599metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 602schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 604metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowUpstreamOperatorsKind].Type; 607schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowUpstreamOperatorsKind, ref inputOps); 614metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 617schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 619metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowUpstreamOperatorsKind].Type; 621schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowUpstreamOperatorsKind, ref inputOps); 629metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 632schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 634metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowUpstreamOperatorsKind].Type; 636schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowUpstreamOperatorsKind, ref inputOps); 641schema = TensorFlowUtils.GetModelSchema(_mlContext, modelLocation); 1134var schema = TensorFlowUtils.GetModelSchema(_mlContext, modelLocation);
Microsoft.ML.Vision (7)
DnnRetrainTransform.cs (5)
19using static Microsoft.ML.TensorFlow.TensorFlowUtils; 114return new DnnRetrainTransformer(env, TensorFlowUtils.LoadTFSession(env, modelBytes), outputs, inputs, 1043var tensor = TensorFlowUtils.CastDataAndReturnAsTensor(_bufferedData, _tfShape); 1091return TensorFlowUtils.CastDataAndReturnAsTensor(_denseData, _tfShape); 1104var tensor = TensorFlowUtils.CastDataAndReturnAsTensor(_bufferedData, _tfShape);
ImageClassificationTrainer.cs (1)
28using static Microsoft.ML.TensorFlow.TensorFlowUtils;
VisionCatalog.cs (1)
9using static Microsoft.ML.TensorFlow.TensorFlowUtils;