diff --git a/nntrainer/layers/layer_context.h b/nntrainer/layers/layer_context.h index fef7ff6d7..33839a143 100644 --- a/nntrainer/layers/layer_context.h +++ b/nntrainer/layers/layer_context.h @@ -901,21 +901,6 @@ class RunLayerContext { */ std::vector 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 @@ -958,9 +943,6 @@ class RunLayerContext { std::vector outputs; /**< outputs of the layer */ std::vector tensors; /**< tensors of the layer */ - ml::train::LayerComputeEngine compute_engine = - ml::train::LayerComputeEngine::CPU; - #ifdef DEBUG std::map tensor_map; /**< map of tensor name to tensor address */