arm_compute v17.10

[platform/upstream/armcl.git] / arm_compute / graph / nodes / SoftmaxLayer.h

diff --git a/arm_compute/graph/nodes/SoftmaxLayer.h b/arm_compute/graph/nodes/SoftmaxLayer.h
index 1779ada..2e1bd98 100644 (file)
--- a/arm_compute/graph/nodes/SoftmaxLayer.h
+++ b/arm_compute/graph/nodes/SoftmaxLayer.h
@@ -24,10 +24,10 @@
 #ifndef __ARM_COMPUTE_GRAPH_SOFTMAX_LAYER_H__
 #define __ARM_COMPUTE_GRAPH_SOFTMAX_LAYER_H__
 
+#include "arm_compute/graph/GraphContext.h"
 #include "arm_compute/graph/INode.h"
 #include "arm_compute/graph/Tensor.h"
 #include "arm_compute/graph/Types.h"
-
 namespace arm_compute
 {
 namespace graph
@@ -37,8 +37,7 @@ class SoftmaxLayer : public INode
 {
 public:
     // Inherited methods overriden:
-    std::unique_ptr<arm_compute::IFunction> instantiate_node(Hint hint, ITensor *input, ITensor *output) override;
-    void print_info() override;
+    std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensor *input, ITensor *output) override;
 };
 
 } // namespace graph