@@ -9,7 +9,7 @@ | |||
[](https://996.icu/#/en_US) | |||
[](https://mybinder.org/v2/gh/javiercp/BinderTF.NET/master?urlpath=lab) | |||
*master branch is based on tensorflow v2.4, v0.3x branch is based on tensorflow v2.3, v0.15-tensorflow1.15 is from tensorflow1.15.* | |||
*master branch is based on tensorflow v2.x, v0.6x branch is based on tensorflow v2.6, v0.15-tensorflow1.15 is from tensorflow1.15.* | |||
 | |||
@@ -249,10 +249,6 @@ Follow us on [Twitter](https://twitter.com/ScisharpStack), [Facebook](https://ww | |||
Join our chat on [Gitter](https://gitter.im/sci-sharp/community). | |||
Scan QR code to join Tencent TIM group: | |||
 | |||
WeChat Sponsor 微信打赏: | |||
 | |||
@@ -0,0 +1,91 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Runtime.InteropServices; | |||
using System.Text; | |||
using Tensorflow.Lite; | |||
namespace Tensorflow | |||
{ | |||
public class c_api_lite | |||
{ | |||
public const string TensorFlowLibName = "tensorflowlite_c"; | |||
public static string StringPiece(IntPtr handle) | |||
{ | |||
return handle == IntPtr.Zero ? String.Empty : Marshal.PtrToStringAnsi(handle); | |||
} | |||
[DllImport(TensorFlowLibName)] | |||
public static extern IntPtr TfLiteVersion(); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern SafeTfLiteModelHandle TfLiteModelCreateFromFile(string model_path); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern void TfLiteModelDelete(IntPtr model); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern SafeTfLiteInterpreterOptionsHandle TfLiteInterpreterOptionsCreate(); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern void TfLiteInterpreterOptionsDelete(IntPtr options); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern void TfLiteInterpreterOptionsSetNumThreads(SafeTfLiteInterpreterOptionsHandle options, int num_threads); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern SafeTfLiteInterpreterHandle TfLiteInterpreterCreate(SafeTfLiteModelHandle model, SafeTfLiteInterpreterOptionsHandle optional_options); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern void TfLiteInterpreterDelete(IntPtr interpreter); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteStatus TfLiteInterpreterAllocateTensors(SafeTfLiteInterpreterHandle interpreter); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern int TfLiteInterpreterGetInputTensorCount(SafeTfLiteInterpreterHandle interpreter); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern int TfLiteInterpreterGetOutputTensorCount(SafeTfLiteInterpreterHandle interpreter); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteStatus TfLiteInterpreterResizeInputTensor(SafeTfLiteInterpreterHandle interpreter, | |||
int input_index, int[] input_dims, int input_dims_size); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteTensor TfLiteInterpreterGetInputTensor(SafeTfLiteInterpreterHandle interpreter, int input_index); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteDataType TfLiteTensorType(TfLiteTensor tensor); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern int TfLiteTensorNumDims(TfLiteTensor tensor); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern int TfLiteTensorDim(TfLiteTensor tensor, int dim_index); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern int TfLiteTensorByteSize(TfLiteTensor tensor); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern IntPtr TfLiteTensorData(TfLiteTensor tensor); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern IntPtr TfLiteTensorName(TfLiteTensor tensor); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteQuantizationParams TfLiteTensorQuantizationParams(TfLiteTensor tensor); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteStatus TfLiteTensorCopyFromBuffer(TfLiteTensor tensor, IntPtr input_data, int input_data_size); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteStatus TfLiteInterpreterInvoke(SafeTfLiteInterpreterHandle interpreter); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern IntPtr TfLiteInterpreterGetOutputTensor(SafeTfLiteInterpreterHandle interpreter, int output_index); | |||
[DllImport(TensorFlowLibName)] | |||
public static extern TfLiteStatus TfLiteTensorCopyToBuffer(TfLiteTensor output_tensor, IntPtr output_data, int output_data_size); | |||
} | |||
} |
@@ -0,0 +1,26 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Text; | |||
using Tensorflow.Util; | |||
namespace Tensorflow.Lite | |||
{ | |||
public class SafeTfLiteInterpreterHandle : SafeTensorflowHandle | |||
{ | |||
protected SafeTfLiteInterpreterHandle() | |||
{ | |||
} | |||
public SafeTfLiteInterpreterHandle(IntPtr handle) | |||
: base(handle) | |||
{ | |||
} | |||
protected override bool ReleaseHandle() | |||
{ | |||
c_api_lite.TfLiteInterpreterDelete(handle); | |||
SetHandle(IntPtr.Zero); | |||
return true; | |||
} | |||
} | |||
} |
@@ -0,0 +1,26 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Text; | |||
using Tensorflow.Util; | |||
namespace Tensorflow.Lite | |||
{ | |||
public class SafeTfLiteInterpreterOptionsHandle : SafeTensorflowHandle | |||
{ | |||
protected SafeTfLiteInterpreterOptionsHandle() | |||
{ | |||
} | |||
public SafeTfLiteInterpreterOptionsHandle(IntPtr handle) | |||
: base(handle) | |||
{ | |||
} | |||
protected override bool ReleaseHandle() | |||
{ | |||
c_api_lite.TfLiteInterpreterOptionsDelete(handle); | |||
SetHandle(IntPtr.Zero); | |||
return true; | |||
} | |||
} | |||
} |
@@ -0,0 +1,26 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Text; | |||
using Tensorflow.Util; | |||
namespace Tensorflow.Lite | |||
{ | |||
public class SafeTfLiteModelHandle : SafeTensorflowHandle | |||
{ | |||
protected SafeTfLiteModelHandle() | |||
{ | |||
} | |||
public SafeTfLiteModelHandle(IntPtr handle) | |||
: base(handle) | |||
{ | |||
} | |||
protected override bool ReleaseHandle() | |||
{ | |||
c_api_lite.TfLiteModelDelete(handle); | |||
SetHandle(IntPtr.Zero); | |||
return true; | |||
} | |||
} | |||
} |
@@ -0,0 +1,27 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Text; | |||
namespace Tensorflow.Lite | |||
{ | |||
public enum TfLiteDataType | |||
{ | |||
kTfLiteNoType = 0, | |||
kTfLiteFloat32 = 1, | |||
kTfLiteInt32 = 2, | |||
kTfLiteUInt8 = 3, | |||
kTfLiteInt64 = 4, | |||
kTfLiteString = 5, | |||
kTfLiteBool = 6, | |||
kTfLiteInt16 = 7, | |||
kTfLiteComplex64 = 8, | |||
kTfLiteInt8 = 9, | |||
kTfLiteFloat16 = 10, | |||
kTfLiteFloat64 = 11, | |||
kTfLiteComplex128 = 12, | |||
kTfLiteUInt64 = 13, | |||
kTfLiteResource = 14, | |||
kTfLiteVariant = 15, | |||
kTfLiteUInt32 = 16, | |||
} | |||
} |
@@ -0,0 +1,12 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Text; | |||
namespace Tensorflow.Lite | |||
{ | |||
public struct TfLiteQuantizationParams | |||
{ | |||
public float scale; | |||
public int zero_point; | |||
} | |||
} |
@@ -0,0 +1,31 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Text; | |||
namespace Tensorflow.Lite | |||
{ | |||
public enum TfLiteStatus | |||
{ | |||
kTfLiteOk = 0, | |||
// Generally referring to an error in the runtime (i.e. interpreter) | |||
kTfLiteError = 1, | |||
// Generally referring to an error from a TfLiteDelegate itself. | |||
kTfLiteDelegateError = 2, | |||
// Generally referring to an error in applying a delegate due to | |||
// incompatibility between runtime and delegate, e.g., this error is returned | |||
// when trying to apply a TfLite delegate onto a model graph that's already | |||
// immutable. | |||
kTfLiteApplicationError = 3, | |||
// Generally referring to serialized delegate data not being found. | |||
// See tflite::delegates::Serialization. | |||
kTfLiteDelegateDataNotFound = 4, | |||
// Generally referring to data-writing issues in delegate serialization. | |||
// See tflite::delegates::Serialization. | |||
kTfLiteDelegateDataWriteError = 5, | |||
} | |||
} |
@@ -0,0 +1,21 @@ | |||
using System; | |||
namespace Tensorflow.Lite | |||
{ | |||
public struct TfLiteTensor | |||
{ | |||
IntPtr _handle; | |||
public TfLiteTensor(IntPtr handle) | |||
=> _handle = handle; | |||
public static implicit operator TfLiteTensor(IntPtr handle) | |||
=> new TfLiteTensor(handle); | |||
public static implicit operator IntPtr(TfLiteTensor tensor) | |||
=> tensor._handle; | |||
public override string ToString() | |||
=> $"TfLiteTensor 0x{_handle.ToString("x16")}"; | |||
} | |||
} |
@@ -5,7 +5,7 @@ | |||
<AssemblyName>TensorFlow.NET</AssemblyName> | |||
<RootNamespace>Tensorflow</RootNamespace> | |||
<TargetTensorFlow>2.2.0</TargetTensorFlow> | |||
<Version>0.60.3</Version> | |||
<Version>0.60.4</Version> | |||
<LangVersion>9.0</LangVersion> | |||
<Nullable>enable</Nullable> | |||
<Authors>Haiping Chen, Meinrad Recheis, Eli Belash</Authors> | |||
@@ -20,7 +20,7 @@ | |||
<Description>Google's TensorFlow full binding in .NET Standard. | |||
Building, training and infering deep learning models. | |||
https://tensorflownet.readthedocs.io</Description> | |||
<AssemblyVersion>0.60.3.0</AssemblyVersion> | |||
<AssemblyVersion>0.60.4.0</AssemblyVersion> | |||
<PackageReleaseNotes>tf.net 0.60.x and above are based on tensorflow native 2.6.0 | |||
* Eager Mode is added finally. | |||
@@ -35,7 +35,7 @@ Keras API is a separate package released as TensorFlow.Keras. | |||
tf.net 0.4x.x aligns with TensorFlow v2.4.1 native library. | |||
tf.net 0.5x.x aligns with TensorFlow v2.5.x native library. | |||
tf.net 0.6x.x aligns with TensorFlow v2.6.x native library.</PackageReleaseNotes> | |||
<FileVersion>0.60.3.0</FileVersion> | |||
<FileVersion>0.60.4.0</FileVersion> | |||
<PackageLicenseFile>LICENSE</PackageLicenseFile> | |||
<PackageRequireLicenseAcceptance>true</PackageRequireLicenseAcceptance> | |||
<SignAssembly>true</SignAssembly> | |||
@@ -189,7 +189,10 @@ namespace Tensorflow | |||
{ | |||
TF_DataType.TF_STRING => "string", | |||
TF_DataType.TF_UINT8 => "uint8", | |||
TF_DataType.TF_INT8 => "int8", | |||
TF_DataType.TF_UINT32 => "uint32", | |||
TF_DataType.TF_INT32 => "int32", | |||
TF_DataType.TF_UINT64 => "uint64", | |||
TF_DataType.TF_INT64 => "int64", | |||
TF_DataType.TF_FLOAT => "float32", | |||
TF_DataType.TF_DOUBLE => "float64", | |||
@@ -204,9 +207,12 @@ namespace Tensorflow | |||
{ | |||
TF_DataType.TF_BOOL => sizeof(bool), | |||
TF_DataType.TF_UINT8 => sizeof(byte), | |||
TF_DataType.TF_INT8 => sizeof(byte), | |||
TF_DataType.TF_INT8 => sizeof(sbyte), | |||
TF_DataType.TF_UINT16 => sizeof(ushort), | |||
TF_DataType.TF_INT16 => sizeof(short), | |||
TF_DataType.TF_UINT32 => sizeof(uint), | |||
TF_DataType.TF_INT32 => sizeof(int), | |||
TF_DataType.TF_UINT64 => sizeof(ulong), | |||
TF_DataType.TF_INT64 => sizeof(long), | |||
TF_DataType.TF_FLOAT => sizeof(float), | |||
TF_DataType.TF_DOUBLE => sizeof(double), | |||
@@ -11,7 +11,9 @@ using static Tensorflow.Binding; | |||
namespace Tensorflow.Benchmark.Leak | |||
{ | |||
/// <summary> | |||
/// https://github.com/SciSharp/TensorFlow.NET/issues/418 | |||
/// </summary> | |||
public class SavedModelCleanup | |||
{ | |||
[Benchmark] | |||
@@ -36,7 +36,7 @@ | |||
</ItemGroup> | |||
<ItemGroup> | |||
<PackageReference Include="BenchmarkDotNet" Version="0.13.0" /> | |||
<PackageReference Include="BenchmarkDotNet" Version="0.13.1" /> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0" /> | |||
</ItemGroup> | |||
@@ -24,14 +24,14 @@ | |||
</PropertyGroup> | |||
<ItemGroup> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.10.0" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.5" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.5" /> | |||
<PackageReference Include="coverlet.collector" Version="3.0.3"> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.7" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.7" /> | |||
<PackageReference Include="coverlet.collector" Version="3.1.0"> | |||
<PrivateAssets>all</PrivateAssets> | |||
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets> | |||
</PackageReference> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0-rc0" /> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0" /> | |||
</ItemGroup> | |||
<ItemGroup> | |||
@@ -14,14 +14,14 @@ | |||
</PropertyGroup> | |||
<ItemGroup> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0-release-20210626-04" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.5" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.5" /> | |||
<PackageReference Include="coverlet.collector" Version="3.0.3"> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.7" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.7" /> | |||
<PackageReference Include="coverlet.collector" Version="3.1.0"> | |||
<PrivateAssets>all</PrivateAssets> | |||
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets> | |||
</PackageReference> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0-rc0" /> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0" /> | |||
</ItemGroup> | |||
<ItemGroup> | |||
@@ -0,0 +1,138 @@ | |||
using Microsoft.VisualStudio.TestTools.UnitTesting; | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Linq; | |||
using System.Runtime.InteropServices; | |||
using System.Text; | |||
using System.Threading.Tasks; | |||
using Tensorflow.Lite; | |||
namespace Tensorflow.Native.UnitTest | |||
{ | |||
[TestClass] | |||
public class TfLiteTest | |||
{ | |||
[TestMethod] | |||
public void TfLiteVersion() | |||
{ | |||
var ver = c_api_lite.StringPiece(c_api_lite.TfLiteVersion()); | |||
Assert.IsNotNull(ver); | |||
} | |||
[TestMethod] | |||
public unsafe void SmokeTest() | |||
{ | |||
var model = c_api_lite.TfLiteModelCreateFromFile("Lite/testdata/add.bin"); | |||
var options = c_api_lite.TfLiteInterpreterOptionsCreate(); | |||
c_api_lite.TfLiteInterpreterOptionsSetNumThreads(options, 2); | |||
var interpreter = c_api_lite.TfLiteInterpreterCreate(model, options); | |||
c_api_lite.TfLiteInterpreterOptionsDelete(options.DangerousGetHandle()); | |||
c_api_lite.TfLiteModelDelete(model.DangerousGetHandle()); | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, c_api_lite.TfLiteInterpreterAllocateTensors(interpreter)); | |||
Assert.AreEqual(1, c_api_lite.TfLiteInterpreterGetInputTensorCount(interpreter)); | |||
Assert.AreEqual(1, c_api_lite.TfLiteInterpreterGetOutputTensorCount(interpreter)); | |||
var input_dims = new int[] { 2 }; | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, c_api_lite.TfLiteInterpreterResizeInputTensor(interpreter, 0, input_dims, input_dims.Length)); | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, c_api_lite.TfLiteInterpreterAllocateTensors(interpreter)); | |||
var input_tensor = c_api_lite.TfLiteInterpreterGetInputTensor(interpreter, 0); | |||
Assert.AreEqual(TfLiteDataType.kTfLiteFloat32, c_api_lite.TfLiteTensorType(input_tensor)); | |||
Assert.AreEqual(1, c_api_lite.TfLiteTensorNumDims(input_tensor)); | |||
Assert.AreEqual(2, c_api_lite.TfLiteTensorDim(input_tensor, 0)); | |||
Assert.AreEqual(sizeof(float) * 2, c_api_lite.TfLiteTensorByteSize(input_tensor)); | |||
Assert.IsNotNull(c_api_lite.TfLiteTensorData(input_tensor)); | |||
Assert.AreEqual("input", c_api_lite.StringPiece(c_api_lite.TfLiteTensorName(input_tensor))); | |||
var input_params = c_api_lite.TfLiteTensorQuantizationParams(input_tensor); | |||
Assert.AreEqual(0f, input_params.scale); | |||
Assert.AreEqual(0, input_params.zero_point); | |||
var input = new[] { 1f, 3f }; | |||
fixed (float* addr = &input[0]) | |||
{ | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, | |||
c_api_lite.TfLiteTensorCopyFromBuffer(input_tensor, new IntPtr(addr), 2 * sizeof(float))); | |||
} | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, c_api_lite.TfLiteInterpreterInvoke(interpreter)); | |||
var output_tensor = c_api_lite.TfLiteInterpreterGetOutputTensor(interpreter, 0); | |||
Assert.AreEqual(TfLiteDataType.kTfLiteFloat32, c_api_lite.TfLiteTensorType(output_tensor)); | |||
Assert.AreEqual(1, c_api_lite.TfLiteTensorNumDims(output_tensor)); | |||
Assert.AreEqual(2, c_api_lite.TfLiteTensorDim(output_tensor, 0)); | |||
Assert.AreEqual(sizeof(float) * 2, c_api_lite.TfLiteTensorByteSize(output_tensor)); | |||
Assert.IsNotNull(c_api_lite.TfLiteTensorData(output_tensor)); | |||
Assert.AreEqual("output", c_api_lite.StringPiece(c_api_lite.TfLiteTensorName(output_tensor))); | |||
var output_params = c_api_lite.TfLiteTensorQuantizationParams(output_tensor); | |||
Assert.AreEqual(0f, output_params.scale); | |||
Assert.AreEqual(0, output_params.zero_point); | |||
var output = new float[2]; | |||
fixed (float* addr = &output[0]) | |||
{ | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, | |||
c_api_lite.TfLiteTensorCopyToBuffer(output_tensor, new IntPtr(addr), 2 * sizeof(float))); | |||
} | |||
Assert.AreEqual(3f, output[0]); | |||
Assert.AreEqual(9f, output[1]); | |||
c_api_lite.TfLiteInterpreterDelete(interpreter.DangerousGetHandle()); | |||
} | |||
[TestMethod] | |||
public unsafe void QuantizationParamsTest() | |||
{ | |||
var model = c_api_lite.TfLiteModelCreateFromFile("Lite/testdata/add_quantized.bin"); | |||
var interpreter = c_api_lite.TfLiteInterpreterCreate(model, new SafeTfLiteInterpreterOptionsHandle(IntPtr.Zero)); | |||
c_api_lite.TfLiteModelDelete(model.DangerousGetHandle()); | |||
var input_dims = new[] { 2 }; | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, c_api_lite.TfLiteInterpreterResizeInputTensor(interpreter, 0, input_dims, 1)); | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, c_api_lite.TfLiteInterpreterAllocateTensors(interpreter)); | |||
var input_tensor = c_api_lite.TfLiteInterpreterGetInputTensor(interpreter, 0); | |||
Assert.IsNotNull(input_tensor); | |||
Assert.AreEqual(TfLiteDataType.kTfLiteUInt8, c_api_lite.TfLiteTensorType(input_tensor)); | |||
Assert.AreEqual(1, c_api_lite.TfLiteTensorNumDims(input_tensor)); | |||
Assert.AreEqual(2, c_api_lite.TfLiteTensorDim(input_tensor, 0)); | |||
var input_params = c_api_lite.TfLiteTensorQuantizationParams(input_tensor); | |||
Assert.AreEqual((0.003922f, 0), (input_params.scale, input_params.zero_point)); | |||
var input = new byte[] { 1, 3 }; | |||
fixed (byte* addr = &input[0]) | |||
{ | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, | |||
c_api_lite.TfLiteTensorCopyFromBuffer(input_tensor, new IntPtr(addr), 2 * sizeof(byte))); | |||
} | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, c_api_lite.TfLiteInterpreterInvoke(interpreter)); | |||
var output_tensor = c_api_lite.TfLiteInterpreterGetOutputTensor(interpreter, 0); | |||
Assert.IsNotNull(output_tensor); | |||
var output_params = c_api_lite.TfLiteTensorQuantizationParams(output_tensor); | |||
Assert.AreEqual((0.003922f, 0), (output_params.scale, output_params.zero_point)); | |||
var output = new byte[2]; | |||
fixed (byte* addr = &output[0]) | |||
{ | |||
Assert.AreEqual(TfLiteStatus.kTfLiteOk, | |||
c_api_lite.TfLiteTensorCopyToBuffer(output_tensor, new IntPtr(addr), 2 * sizeof(byte))); | |||
} | |||
Assert.AreEqual(3f, output[0]); | |||
Assert.AreEqual(9f, output[1]); | |||
var dequantizedOutput0 = output_params.scale * (output[0] - output_params.zero_point); | |||
var dequantizedOutput1 = output_params.scale * (output[1] - output_params.zero_point); | |||
Assert.AreEqual(dequantizedOutput0, 0.011766f); | |||
Assert.AreEqual(dequantizedOutput1, 0.035298f); | |||
c_api_lite.TfLiteInterpreterDelete(interpreter.DangerousGetHandle()); | |||
} | |||
} | |||
} |
@@ -24,16 +24,31 @@ | |||
<AllowUnsafeBlocks>true</AllowUnsafeBlocks> | |||
</PropertyGroup> | |||
<ItemGroup> | |||
<None Remove="Lite\testdata\add.bin" /> | |||
<None Remove="Lite\testdata\add_quantized.bin" /> | |||
</ItemGroup> | |||
<ItemGroup> | |||
<Content Include="Lite\testdata\add.bin"> | |||
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory> | |||
</Content> | |||
<Content Include="Lite\testdata\add_quantized.bin"> | |||
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory> | |||
</Content> | |||
</ItemGroup> | |||
<ItemGroup> | |||
<PackageReference Include="FluentAssertions" Version="5.10.3" /> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0-release-20210626-04" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.5" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.5" /> | |||
<PackageReference Include="coverlet.collector" Version="3.0.3"> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.7" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.7" /> | |||
<PackageReference Include="coverlet.collector" Version="3.1.0"> | |||
<PrivateAssets>all</PrivateAssets> | |||
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets> | |||
</PackageReference> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0-rc0" /> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0" /> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist-Lite" Version="2.6.0" /> | |||
</ItemGroup> | |||
<ItemGroup> | |||
@@ -48,10 +48,10 @@ | |||
<ItemGroup> | |||
<PackageReference Include="FluentAssertions" Version="5.10.3" /> | |||
<PackageReference Include="MethodBoundaryAspect.Fody" Version="2.0.139" /> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0-release-20210626-04" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.5" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.5" /> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0-rc0" /> | |||
<PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0" /> | |||
<PackageReference Include="MSTest.TestAdapter" Version="2.2.7" /> | |||
<PackageReference Include="MSTest.TestFramework" Version="2.2.7" /> | |||
<PackageReference Include="SciSharp.TensorFlow.Redist" Version="2.6.0" /> | |||
</ItemGroup> | |||
<ItemGroup> | |||