2 #error "Define DATA_CHEF first"
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)
25 // FLOAT16 support for only gaussian, explicit for now
26 DATA_CHEF(FLOAT16, explicit, ExplicitFloat16DataChefFactory)
27 DATA_CHEF(FLOAT16, gaussian, GaussianFloat16DataChefFactory)