target_include_directories(tflchef_core PRIVATE ${TensorFlowSource_DIR})
target_include_directories(tflchef_core PRIVATE src)
target_link_libraries(tflchef_core tflchef_proto)
-target_link_libraries(tflchef_core stdex)
target_link_libraries(tflchef_core flatbuffers)
#include "Dataset.h"
-#include <stdex/Memory.h>
-
#include <map>
#include <string>
#include <vector>
#include <numeric>
#include <stdexcept>
-using stdex::make_unique;
-
namespace
{
GeneratedModel cook(const ::tflchef::ModelRecipe &model_recipe)
{
// Initialize Op Chef Registry
-#define OP_CHEF(NAME, FACTORY_CLASS) op_chef_registry().add(#NAME, make_unique<FACTORY_CLASS>());
+#define OP_CHEF(NAME, FACTORY_CLASS) \
+ op_chef_registry().add(#NAME, std::unique_ptr<FACTORY_CLASS>(new FACTORY_CLASS()));
#include "OpChef.def"
#undef OP_CHEF
// Initialize Data Chef Registry
#define DATA_CHEF(TYPE, NAME, FACTORY_CLASS) \
- data_chef_registry(::tflchef::TYPE).add(#NAME, make_unique<FACTORY_CLASS>());
+ data_chef_registry(::tflchef::TYPE) \
+ .add(#NAME, std::unique_ptr<FACTORY_CLASS>(new FACTORY_CLASS()));
#include "DataChef.def"
#undef DATA_CHEF
//
// Create FlatBufferBuilder
//
- auto flatbuffer_builder = make_unique<flatbuffers::FlatBufferBuilder>(1024);
+ auto flatbuffer_builder =
+ std::unique_ptr<flatbuffers::FlatBufferBuilder>(new flatbuffers::FlatBufferBuilder(1024));
// Operand-related
std::vector<flatbuffers::Offset<::tflite::Buffer>> buffer_vec;
::tflite::FinishModelBuffer(*flatbuffer_builder, model);
// Return "GenerateModel"
- return GeneratedModel{make_unique<GeneratedModelImpl>(std::move(flatbuffer_builder))};
+ return GeneratedModel{
+ std::unique_ptr<GeneratedModelImpl>(new GeneratedModelImpl(std::move(flatbuffer_builder)))};
}
} // namespace tflchef