41 references to TensorFlowUtils
Microsoft.ML.TensorFlow (15)
TensorflowCatalog.cs (2)
38=> TensorFlowUtils.LoadTensorFlowModel(CatalogUtils.GetEnvironment(catalog), modelLocation); 65=> 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)
564var schema = TensorFlowUtils.GetModelSchema(_mlContext, modelLocation); 571var metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 575schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 577metadataType = schema[col].Annotations.Schema.GetColumnOrNull(TensorFlowUtils.TensorflowUpstreamOperatorsKind)?.Type; 583metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 586schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 588metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowUpstreamOperatorsKind].Type; 591schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowUpstreamOperatorsKind, ref inputOps); 598metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 601schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 603metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowUpstreamOperatorsKind].Type; 605schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowUpstreamOperatorsKind, ref inputOps); 613metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowOperatorTypeKind].Type; 616schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowOperatorTypeKind, ref opType); 618metadataType = schema[col].Annotations.Schema[TensorFlowUtils.TensorflowUpstreamOperatorsKind].Type; 620schema[col].Annotations.GetValue(TensorFlowUtils.TensorflowUpstreamOperatorsKind, ref inputOps); 625schema = TensorFlowUtils.GetModelSchema(_mlContext, modelLocation); 1118var 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;