diff --git a/mace/core/mace.cc b/mace/core/mace.cc index 0d54c89563210e02dee44538f3164ff23e89943c..a533ff0513f7de3b71c7660c84fd965586913b9e 100644 --- a/mace/core/mace.cc +++ b/mace/core/mace.cc @@ -556,6 +556,13 @@ MaceEngine::~MaceEngine() { MACE_CHECK(hexagon_controller_->Finalize(), "hexagon finalize error"); } }; + +bool MaceEngine::Run(const float *input, + const std::vector &input_shape, + float *output) { + return Run(input, input_shape, output, nullptr); +} + bool MaceEngine::Run(const float *input, const std::vector &input_shape, float *output, diff --git a/mace/core/public/mace.h b/mace/core/public/mace.h index e7036ffe00cbff3ff0eb20a7617a036a2e36d33f..23fe80040e31e5f949dc1ed738b96d9521242fd9 100644 --- a/mace/core/public/mace.h +++ b/mace/core/public/mace.h @@ -360,10 +360,13 @@ class MaceEngine { explicit MaceEngine(const NetDef *net_def, DeviceType device_type); ~MaceEngine(); + bool Run(const float *input, + const std::vector &input_shape, + float *output); bool Run(const float *input, const std::vector &input_shape, float *output, - RunMetadata *run_metadata = nullptr); + RunMetadata *run_metadata); MaceEngine(const MaceEngine &) = delete; MaceEngine &operator=(const MaceEngine &) = delete;