Packages

tfl_interp

0.1.13
0.1.16 0.1.15 0.1.14 0.1.13 0.1.12 0.1.11 0.1.10 0.1.9 0.1.8 0.1.7 0.1.6 0.1.5 0.1.4 0.1.3 0.1.2 0.1.1

Tensorflow lite intepreter for Elixir.

HexDocs

Current section

Files

Jump to
Readme
16 Versions
4 Dependencies
Dependants
Files
Activity
tfl_interp src custom_ops custom_operations.cc
Raw

src/custom_ops/custom_operations.cc 

/***  File Header  ************************************************************/

/**


* custom_operations.cc


*


* Elixir/Erlang Port ext. of tensor flow lite


* @author	   Shozo Fukuda


* @date	create Sat Apr 15 17:12:44 2023


* System	   VC++ 2019/Windows 10<br>


*


**/

/**************************************************************************{{{*/



#include "tensorflow/lite/kernels/register.h"



#include "max_pool_argmax.h"

#include "max_unpooling.h"

#include "transpose_conv_bias.h"

#include "extract_image_patches.h"



/***  Module Header  ******************************************************}}}*/

/**


* install custom operations


* @par DESCRIPTION


*


*


* @retval


**/

/**************************************************************************{{{*/

void add_custom_operations(tflite::ops::builtin::BuiltinOpResolver& resolver)

{

    resolver.AddCustom("MaxPoolingWithArgmax2D", custom_operations::RegisterMaxPoolingWithArgmax2D());

    resolver.AddCustom("MaxUnpooling2D", custom_operations::RegisterMaxUnpooling2D());

    resolver.AddCustom("Convolution2DTransposeBias", custom_operations::RegisterConvolution2DTransposeBias());

    resolver.AddCustom("ExtractImagePatches", custom_operations::RegisterExtractImagePatches());

}



/*** custom_operations.cc *************************************************}}}*/