Tensor: added a constructor to create a Tensor from a fixed memory pointer that is owned by the caller

Author: henon

TensorFlow.NET.sln

@@ -1,7 +1,7 @@
 
 Microsoft Visual Studio Solution File, Format Version 12.00
-# Visual Studio Version 16
-VisualStudioVersion = 16.0.29102.190
+# Visual Studio 15
+VisualStudioVersion = 15.0.28307.645
 MinimumVisualStudioVersion = 10.0.40219.1
 Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "TensorFlowNET.UnitTest", "test\TensorFlowNET.UnitTest\TensorFlowNET.UnitTest.csproj", "{029A8CF1-CF95-4DCB-98AA-9D3D96A83B3E}"
 EndProject
@@ -17,7 +17,7 @@ Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "Keras.UnitTest", "test\Kera
 EndProject
 Project("{6EC3EE1D-3C4E-46DD-8F32-0CC8E7565705}") = "TensorFlowNET.Examples.FSharp", "test\TensorFlowNET.Examples.FSharp\TensorFlowNET.Examples.FSharp.fsproj", "{62BC3801-F0D3-44A9-A0AC-712F40C8F961}"
 EndProject
-Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "TensorFlowNET.Benchmark", "src\TensorFlowNet.Benchmarks\TensorFlowNET.Benchmark.csproj", "{68861442-971A-4196-876E-C9330F0B3C54}"
+Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "TensorFlowBenchmark", "src\TensorFlowNet.Benchmarks\TensorFlowBenchmark.csproj", "{68861442-971A-4196-876E-C9330F0B3C54}"
 EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution

src/TensorFlowNET.Core/Tensors/Tensor.Creation.cs

@@ -3,15 +3,10 @@
 using BenchmarkDotNet.Configs;
 using BenchmarkDotNet.Running;
 
-namespace TensorFlowNet.Benchmark
+namespace TensorFlowBenchmark
 {
     class Program
     {
-        /// <summary>
-        /// dotnet NumSharp.Benchmark.dll (Benchmark Class Name)
-        /// dotnet NumSharp.Benchmark.dll nparange
-        /// </summary>
-        /// <param name="args"></param>
         static void Main(string[] args)
         {
 #if DEBUG
@@ -24,7 +19,7 @@ static void Main(string[] args)
             {
                 for (int i = 0; i < args.Length; i++)
                 {
-                    string name = $"TensorFlowNet.Benchmark.{args[i]}";
+                    string name = $"TensorFlowBenchmark.{args[i]}";
                     var type = Type.GetType(name);
                     BenchmarkRunner.Run(type, config);
                 }

src/TensorFlowNet.Benchmarks/Program.cs

@@ -1,11 +1,9 @@
 using System;
-using System.Linq;
 using BenchmarkDotNet.Attributes;
-using BenchmarkDotNet.Engines;
 using NumSharp;
 using Tensorflow;
 
-namespace TensorFlowNet.Benchmark
+namespace TensorFlowBenchmark
 {
     [SimpleJob(launchCount: 1, warmupCount: 2, targetCount: 10)]
     [MinColumn, MaxColumn, MeanColumn, MedianColumn]
@@ -33,15 +31,16 @@ public void ScalarTensor()
         }
 
         [Benchmark]
-        public void TensorFromSpan()
+        public unsafe void TensorFromFixedPtr()
         {
             var g = new Graph();
-            var span = new Span<double>(data);
             for (int i = 0; i < 100; i++)
             {
-                using (var tensor = new Tensor(span, new long[] { data.Length }))
+                fixed (double* ptr = &data[0])
                 {
-
+                    using (var t = new Tensor((IntPtr)ptr, new long[] { data.Length }, tf.float64, 8 * data.Length))
+                    {
+                    }
                 }
             }
         }

src/TensorFlowNet.Benchmarks/TensorBenchmark.cs

@@ -0,0 +1,31 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+  <PropertyGroup>
+    <OutputType>Exe</OutputType>
+    <TargetFramework>netcoreapp2.2</TargetFramework>
+    <NoWin32Manifest>true</NoWin32Manifest>
+    <AssemblyName>TensorFlowBenchmark</AssemblyName>
+    <RootNamespace>TensorFlowBenchmark</RootNamespace>
+  </PropertyGroup>
+
+  <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|AnyCPU'">
+    <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
+  </PropertyGroup>
+
+  <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|AnyCPU'">
+    <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <None Remove="tensorflow.dll" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <PackageReference Include="BenchmarkDotNet" Version="0.11.5" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <ProjectReference Include="..\TensorFlowNET.Core\TensorFlowNET.Core.csproj" />
+  </ItemGroup>
+
+</Project>

src/TensorFlowNet.Benchmarks/TensorFlowBenchmark.csproj

@@ -53,14 +53,27 @@ public void TensorDeallocationThreadSafety()
         }
 
         [TestMethod]
-        public void TensorFromSpan()
-        {
-            var array = new int[1000];
-            var span = new Span<int>(array, 100, 500);
-            using (var t = new Tensor(span, new long[] { span.Length }))
+        public unsafe void TensorFromFixed()
+        {
+            var array = new float[1000];
+            var span = new Span<float>(array, 100, 500);
+            fixed (float* ptr=&MemoryMarshal.GetReference(span))
+            {
+                using (var t = new Tensor((IntPtr)ptr, new long[] {span.Length}, tf.float32, 4*span.Length))
+                {
+                    Assert.IsFalse(t.IsDisposed);
+                    Assert.IsFalse(t.IsMemoryOwner);
+                    Assert.AreEqual(2000, (int) t.bytesize);
+                }
+            }
+            fixed (float* ptr = &array[0])
             {
-                Assert.IsFalse(t.IsDisposed);
-                Assert.AreEqual(2000, (int)t.bytesize);
+                using (var t = new Tensor((IntPtr)ptr, new long[] { array.Length }, tf.float32, 4 * array.Length))
+                {
+                    Assert.IsFalse(t.IsDisposed);
+                    Assert.IsFalse(t.IsMemoryOwner);
+                    Assert.AreEqual(4000, (int)t.bytesize);
+                }
             }
         }