compiler/tflchef/core/src/DataChef.def

   1 #ifndef DATA_CHEF
   2 #error "Define DATA_CHEF first"
   3 #endif // DATA_CHEF
   4
   5 // DATA_CHEF(TYPE, NAME, FACTORY_CLASS)
   6 //  "TYPE" SHOULD BE an enum tag of tflchef::TensorType
   7 DATA_CHEF(FLOAT32, constant, ConstantDataChefFactory<float>)
   8 DATA_CHEF(BOOL, constant, ConstantDataChefFactory<bool>)
   9 DATA_CHEF(UINT8, constant, ConstantDataChefFactory<uint8_t>)
  10 DATA_CHEF(INT16, constant, ConstantDataChefFactory<int16_t>)
  11 DATA_CHEF(INT32, constant, ConstantDataChefFactory<int32_t>)
  12 DATA_CHEF(INT64, constant, ConstantDataChefFactory<int64_t>)
  13 DATA_CHEF(INT64, explicit, ExplicitDataChefFactory<int64_t>)
  14 DATA_CHEF(INT32, explicit, ExplicitDataChefFactory<int32_t>)
  15 DATA_CHEF(INT16, explicit, ExplicitDataChefFactory<int16_t>)
  16 DATA_CHEF(UINT8, explicit, ExplicitDataChefFactory<uint8_t>)
  17 DATA_CHEF(BOOL, explicit, ExplicitDataChefFactory<bool>)
  18 DATA_CHEF(FLOAT32, explicit, ExplicitDataChefFactory<float>)
  19 DATA_CHEF(STRING, explicit, ExplicitDataChefFactory<std::string>)
  20 DATA_CHEF(FLOAT32, gaussian, GaussianFloat32DataChefFactory)
  21 DATA_CHEF(INT32, gaussian, GaussianInt32DataChefFactory)
  22 DATA_CHEF(INT16, gaussian, GaussianInt16DataChefFactory)
  23 DATA_CHEF(UINT8, gaussian, GaussianUint8DataChefFactory)
  24
  25 // FLOAT16 support for only gaussian, explicit for now
  26 DATA_CHEF(FLOAT16, explicit, ExplicitFloat16DataChefFactory)
  27 DATA_CHEF(FLOAT16, gaussian, GaussianFloat16DataChefFactory)