TensorFlow.NET/src/TensorFlowNET.Core/APIs/c_api.cs at master · chuckbeasley/TensorFlow.NET · GitHub

History

107 lines (92 loc) · 3.72 KB

Raw

100

101

102

103

104

105

106

107

/*****************************************************************************

Licensed under the Apache License, Version 2.0 (the "License");

you may not use this file except in compliance with the License.

You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software

distributed under the License is distributed on an "AS IS" BASIS,

WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

See the License for the specific language governing permissions and

limitations under the License.

******************************************************************************/

using System;

using System.Runtime.InteropServices;

using static Tensorflow.CppShapeInferenceResult.Types;

namespace Tensorflow

{

/// <summary>

/// C API for TensorFlow.

/// Port from tensorflow\c\c_api.h

///

/// The API leans towards simplicity and uniformity instead of convenience

/// since most usage will be by language specific wrappers.

///

/// The params type mapping between c_api and .NET

/// TF_XX** => ref IntPtr (TF_Operation** op) => (ref IntPtr op)

/// TF_XX* => IntPtr (TF_Graph* graph) => (IntPtr graph)

/// struct => struct (TF_Output output) => (TF_Output output)

/// struct* => struct[] (TF_Output* output) => (TF_Output[] output)

/// struct* => struct* for ref

/// const char* => string

/// int32_t => int

/// int64_t* => long[]

/// size_t* => ulong[]

/// size_t* => ref ulong

/// void* => IntPtr

/// string => IntPtr c_api.StringPiece(IntPtr)

/// unsigned char => byte

/// </summary>

public partial class c_api

{

public const string TensorFlowLibName = "tensorflow";

public static string StringPiece(IntPtr handle)

{

return handle == IntPtr.Zero ? String.Empty : Marshal.PtrToStringAnsi(handle);

}

public unsafe static byte[] ByteStringPiece(Buffer? handle)

{

if (handle is null)

{

return new byte[0];

}

var data = handle.ToArray();

return data;

}

public unsafe static byte[] ByteStringPieceFromNativeString(IntPtr handle)

{

if (handle == IntPtr.Zero)

{

return new byte[0];

}

byte* str_data = (byte*)handle.ToPointer();

List<byte> bytes = new List<byte>();

byte current = 255;

while (current != ((byte)'\0'))

{

current = *(str_data++);

bytes.Add(current);

}

var data = bytes.ToArray();

return data;

}

[UnmanagedFunctionPointer(CallingConvention.Winapi)]

public delegate void Deallocator(IntPtr data, IntPtr size, ref DeallocatorArgs args);

[UnmanagedFunctionPointer(CallingConvention.Winapi)]

public delegate void DeallocatorV2(IntPtr data, long size, IntPtr args);

public struct DeallocatorArgs

{

internal static unsafe c_api.DeallocatorArgs* EmptyPtr;

internal static unsafe IntPtr Empty;

static unsafe DeallocatorArgs()

{

Empty = new IntPtr(EmptyPtr = (DeallocatorArgs*)Marshal.AllocHGlobal(Marshal.SizeOf<DeallocatorArgs>()));

*EmptyPtr = new DeallocatorArgs() { gc_handle = IntPtr.Zero, deallocator_called = false };

}

public bool deallocator_called;

public IntPtr gc_handle;

}

[DllImport(TensorFlowLibName)]

internal static extern IntPtr TF_Version();

}

TensorFlow.NET/src/TensorFlowNET.Core/APIs/c_api.cs at master · chuckbeasley/TensorFlow.NET · GitHub

Expand file tree

Breadcrumbs

c_api.cs

Latest commit

History

Breadcrumbs

c_api.cs

File metadata and controls