29 lines
679 B
C++
29 lines
679 B
C++
#pragma once
|
|
#include "PianoInferencer.h"
|
|
|
|
|
|
namespace dynasty {
|
|
namespace inferencer {
|
|
namespace fix_point {
|
|
namespace kl720 {
|
|
|
|
|
|
|
|
class InferencerImplWrapperFactory {
|
|
public:
|
|
static InferencerUniquePtr<float> create(std::string model_file, bool cuda_enabled, int device_id=0);
|
|
static InferencerUniquePtr<float> create(std::string model_file, std::string jsonFile, bool cuda_enabled, int device_id=0);
|
|
|
|
|
|
// Get out the builder which is used to build the inferencer
|
|
static std::shared_ptr<PianoInferencer<int>::Builder> GetBuilder(AbstractInferencer<float>* inferencer);
|
|
};
|
|
|
|
|
|
} // End of kl720
|
|
} // End of fix_point
|
|
} // End of inferencer
|
|
} // End of dynasty
|
|
|
|
|