diff options
Diffstat (limited to 'arm_compute/graph2/frontend')
-rw-r--r-- | arm_compute/graph2/frontend/Stream.h | 10 | ||||
-rw-r--r-- | arm_compute/graph2/frontend/Types.h | 1 |
2 files changed, 4 insertions, 7 deletions
diff --git a/arm_compute/graph2/frontend/Stream.h b/arm_compute/graph2/frontend/Stream.h index 6100975958..bfefe12225 100644 --- a/arm_compute/graph2/frontend/Stream.h +++ b/arm_compute/graph2/frontend/Stream.h @@ -61,14 +61,10 @@ public: Stream &operator=(Stream &&) = default; /** Finalizes the stream for an execution target * - * @note enable_tuning only works if the target is OpenCL. - * @note tuning increases the execution time of first run of the graph - * - * @param[in] target Execution target - * @param[in] enable_tuning (Optional) Enables the tuning interface. Defaults to false - * @param[in] enable_memory_management (Optional) Enables the memory management interface. Defaults to false + * @param[in] target Execution target + * @param[in] config (Optional) Graph configuration to use */ - void finalize(Target target, bool enable_tuning = false, bool enable_memory_management = false); + void finalize(Target target, const GraphConfig &config); /** Executes the stream **/ void run(); diff --git a/arm_compute/graph2/frontend/Types.h b/arm_compute/graph2/frontend/Types.h index 234b998126..d433d1547b 100644 --- a/arm_compute/graph2/frontend/Types.h +++ b/arm_compute/graph2/frontend/Types.h @@ -47,6 +47,7 @@ using graph2::ConvolutionMethod; using graph2::DepthwiseConvolutionMethod; using graph2::TensorDescriptor; using graph2::DimensionRoundingType; +using graph2::GraphConfig; /** Branch layer merging method */ enum class BranchMergeMethod |