Backend::Backend(const std::shared_ptr<neurun::backend::IBackendConfig> &backend_config,
const std::shared_ptr<neurun::backend::IInitializerGenerator> &initializer_gen,
const std::shared_ptr<neurun::backend::IStageGenerator> &stage_gen)
- : _initializer_gen(initializer_gen), _stage_gen(stage_gen)
+ : _config(backend_config), _initializer_gen(initializer_gen), _stage_gen(stage_gen)
{
backend_config->initialize();
}
+const std::shared_ptr<neurun::backend::IBackendConfig> Backend::config() const { return _config; }
+
const std::shared_ptr<neurun::backend::IInitializerGenerator> Backend::initializer_gen() const
{
return _initializer_gen;
const std::shared_ptr<neurun::backend::IInitializerGenerator> &initializer_gen,
const std::shared_ptr<neurun::backend::IStageGenerator> &stage_gen);
- Backend(void) : _initializer_gen(nullptr), _stage_gen(nullptr)
+ Backend(void) : _config(nullptr), _initializer_gen(nullptr), _stage_gen(nullptr)
{
// DO NOTHING
}
public:
+ const std::shared_ptr<neurun::backend::IBackendConfig> config() const;
const std::shared_ptr<neurun::backend::IInitializerGenerator> initializer_gen() const;
const std::shared_ptr<neurun::backend::IStageGenerator> stage_gen() const;
const std::shared_ptr<neurun::backend::ITensorBuilder> tensor_builder() const;
private:
+ std::shared_ptr<neurun::backend::IBackendConfig> _config;
std::shared_ptr<neurun::backend::IInitializerGenerator> _initializer_gen;
std::shared_ptr<neurun::backend::IStageGenerator> _stage_gen;
};