diff --git a/nntrainer/layers/layer_context.h b/nntrainer/layers/layer_context.h index fef7ff6d7..2c9f19d2e 100644 --- a/nntrainer/layers/layer_context.h +++ b/nntrainer/layers/layer_context.h @@ -901,21 +901,6 @@ class RunLayerContext { */ std::vector<Weight *> getWeights() { return weights; } - /** - * @brief set the compute engine for this node - * @param compute engine: (CPU/GPU) - */ - void setComputeEngine(const ml::train::LayerComputeEngine &compute_engine = - ml::train::LayerComputeEngine::CPU) { - this->compute_engine = compute_engine; - } - - /** - * @brief get the compute engine for this node - * @return ompute engine: (CPU/GPU) - */ - ml::train::LayerComputeEngine getComputeEngine() { return compute_engine; } - /** * @brief get loss scale * @return loss scale