tf.prg
Code: Select all | Expand
#include "FiveWin.ch"#define TF_STRING 6static hDLLfunction Main() local oTF := TensorFlow():New() local hTensor, hOperationDescription, hOperation local hTensorOutput := 0 MsgInfo( oTF:Version() ) hTensor = oTF:TensorString( "Hello TensorFlow" ) hOperationDescription = TF_NewOperation( oTF:hGraph, "Const", "hello" ) TF_SetAttrTensor( hOperationDescription, "value", hTensor, oTF:hStatus ) TF_SetAttrType( hOperationDescription, "dtype", TF_TensorType( hTensor ) ) hOperation = TF_FinishOperation( hOperationDescription, oTF:hStatus ) TF_SessionRun( oTF:hSession, 0,; 0, 0, 0,; // Inputs TF_Output( hOperation, 0 ), @hTensorOutput, 1,; // Outputs hOperation, 1,; // Operations 0, oTF:hStatus ) MsgInfo( oTF:StatusCode(), "zero means ok" ) MsgInfo( TF_TensorType( hTensorOutput ) ) MsgInfo( L2Str( TF_TensorData( hTensorOutput ) ) ) oTF:End()return nilCLASS TensorFlow DATA hGraph DATA hOptions DATA hStatus DATA hSession METHOD New() METHOD Version() INLINE TF_Version() METHOD ImportGraph( cFileName ) METHOD TensorString( cString ) METHOD End() METHOD StatusCode() INLINE TF_GetCode( ::hStatus ) METHOD Variable( cVarName, nDataType )ENDCLASSMETHOD New() CLASS TensorFlow hDLL = LoadLibrary( "tensorflow.dll" ) ::hGraph = TF_NewGraph() ::hOptions = TF_NewSessionOptions() ::hStatus = TF_NewStatus() ::hSession = TF_NewSession( ::hGraph, ::hOptions, ::hStatus ) return Self METHOD End() CLASS TensorFlow TF_CloseSession( ::hSession, ::hStatus ) TF_DeleteSession( ::hSession, ::hStatus ) TF_DeleteStatus( ::hStatus ) TF_DeleteSessionOptions( ::hOptions ) FreeLibrary( hDLL )return nilMETHOD ImportGraph( cFileName ) CLASS TensorFlow local cGraph := MemoRead( cFileName ) local hGraphDefOptions := TF_NewImportGraphDefOptions() TF_GraphImportGraphDef( ::hGraph, TF_NewBufferFromString( cGraph, Len( cGraph ) ), hGraphDefOptions, ::hStatus ) if TF_GetCode( ::hStatus ) != 0 MsgAlert( TF_Message( ::hStatus ), "Error importing a Graph file" ) endif TF_DeleteImportGraphDefOptions( hGraphDefOptions ) return nil METHOD TensorString( cString ) CLASS TensorFlow local hTensor := TF_AllocateTensor( TF_STRING, 0, 0, 8 + TF_StringEncodedSize( Len( cString ) ) ) Memset( TF_TensorData( hTensor ), 0, 8 ) TF_StringEncode( cString, Len( cString ), 8 + TF_TensorData( hTensor ), TF_StringEncodedSize( Len( cString ) ), ::hStatus ) if TF_GetCode( ::hStatus ) != 0 MsgAlert( TF_Message( ::hStatus ), "Error creating a Tensor string" ) endif return hTensor METHOD Variable( cVarName, nDataType ) CLASS TensorFlow local hOperationDescription := TF_NewOperation( ::hGraph, "Variable", cVarName ) TF_SetAttrType( hOperationDescription, "dtype", nDataType )return TF_FinishOperation( hOperationDescription, ::hStatus )DLL FUNCTION TF_Version() AS LPSTR LIB hDLLDLL FUNCTION TF_NewGraph() AS LONG LIB hDLLDLL FUNCTION TF_NewSessionOptions() AS LONG LIB hDLLDLL FUNCTION TF_NewStatus() AS LONG LIB hDLLDLL FUNCTION TF_NewSession( hGraph AS LONG, hOptions AS LONG, hStatus AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_CloseSession( hSession AS LONG, hStatus AS LONG ) AS VOID LIB hDLLDLL FUNCTION TF_DeleteSession( hSession AS LONG, hStatus AS LONG ) AS VOID LIB hDLLDLL FUNCTION TF_DeleteStatus( hStatus AS LONG ) AS VOID LIB hDLLDLL FUNCTION TF_DeleteSessionOptions( hOptions AS LONG ) AS VOID LIB hDLLDLL FUNCTION TF_NewImportGraphDefOptions() AS LONG LIB hDLLDLL FUNCTION TF_GraphImportGraphDef( hGraph AS LONG, hBuffer AS LONG, hGraphDefOptions AS LONG, hStatus AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_NewBufferFromString( cString AS LPSTR, nLegth AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_GetCode( hStatus AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_DeleteImportGraphDefOptions( hGraphDefOptions AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_NewTensor( nType AS LONG, pDims AS LONG, nDims AS LONG, pData AS LONG, nLength AS LONG, pDeallocator AS LONG,; pDeallocatorArgs AS LONG ) AS LONG LIB hDLL DLL FUNCTION TF_AllocateTensor( nType AS LONG, pDims AS LONG, nDims AS LONG, nLegth AS LONG ) AS LONG LIB hDLL DLL FUNCTION TF_StringEncodedSize( nLength AS LONG ) AS LONG LIB hDLL DLL FUNCTION TF_TensorData( hTensor AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_StringEncode( cString AS LPSTR, nLength AS LONG, pDest AS LONG, nDestLength AS LONG, hStatus AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_SessionRun( hSession AS LONG, hRunOptions AS LONG, hInputs AS LONG, @hInputValues AS LONG, nInputs AS LONG,; hOutputs AS LONG, @hOutputValues AS LONG, nOutputs AS LONG, @hTargetOperations AS LONG, nTargets AS LONG,; hRunMetadata AS LONG, hStatus AS LONG ) AS VOID LIB hDLLDLL FUNCTION TF_Message( hStatus AS LONG ) AS LPSTR LIB hDLLDLL FUNCTION TF_NewOperation( hGraph AS LONG, cOperationType AS LPSTR, cOperationName AS LPSTR ) AS LONG LIB hDLLDLL FUNCTION TF_SetAttrTensor( hOperationDescription AS LONG, cAttributeName AS LPSTR, hTensor AS LONG, hStatus AS LONG ) AS VOID LIB hDLLDLL FUNCTION TF_TensorType( hTensor AS LONG ) AS LONG LIB hDLLDLL FUNCTION TF_SetAttrType( hOperationDescription AS LONG, cAttributeName AS LPSTR, nDataType AS LONG ) AS VOID LIB hDLLDLL FUNCTION TF_FinishOperation( hOperationDescription AS LONG, hStatus AS LONG ) AS LONG LIB hDLL#pragma BEGINDUMP#include <hbapi.h>typedef struct TF_Output { void * oper; int index;} TF_Output;HB_FUNC( TF_OUTPUT ){ TF_Output * hOutput = ( TF_Output * ) hb_xgrab( sizeof( TF_Output ) ); hOutput->oper = ( void * ) hb_parnll( 1 ); hOutput->index = hb_parnl( 2 ); hb_retnll( ( HB_LONGLONG ) hOutput );} HB_FUNC( L2STR ){ hb_retc( ( ( char * ) hb_parnll( 1 ) ) + 9 );} HB_FUNC( MEMSET ){ hb_retnll( ( HB_LONGLONG ) memset( ( void * ) hb_parnll( 1 ), hb_parnl( 2 ), hb_parnll( 3 ) ) );} #pragma ENDDUMP
