Skip to content

Commit

Permalink
make Tanh.h, Tanh.cc
Browse files Browse the repository at this point in the history
  • Loading branch information
timedilation committed Oct 21, 2018
1 parent 99315de commit a560271
Show file tree
Hide file tree
Showing 2 changed files with 110 additions and 0 deletions.
59 changes: 59 additions & 0 deletions newnnfw/runtimes/neurun/src/graph/operation/Tanh.cc
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
/*
* Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#include "Tanh.h"

#include <cassert>

#include "NodeVisitor.h"
#include "LowerInfo.h"

namespace neurun
{
namespace graph
{
namespace operation
{
namespace Tanh
{

void Node::accept(NodeVisitor &&v) const { v.visit(*this); }

Node::Node(const graph::operation::Node::InitParam &init_param)
{
assert(init_param.input_count == 1 && init_param.output_count == 1);
setInputs({init_param.inputs[0]});
setOutputs({init_param.outputs[0]});
}

void Node::setInputs(const operand::IndexSet &indexes)
{
assert(indexes.size() == 1);

graph::operation::Node::setInputs(indexes);
}

void Node::setOutputs(const operand::IndexSet &indexes)
{
assert(indexes.size() == 1);

graph::operation::Node::setOutputs(indexes);
}

} // namespace Tanh
} // namespace operation
} // namespace graph
} // namespace neurun
51 changes: 51 additions & 0 deletions newnnfw/runtimes/neurun/src/graph/operation/Tanh.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
/*
* Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#ifndef __NEURUN_GRAPH_OPERATION_TANH_H__
#define __NEURUN_GRAPH_OPERATION_TANH_H__

#include <memory>

#include "graph/operation/Node.h"

namespace neurun
{
namespace graph
{
namespace operation
{
namespace Tanh
{

class Node : public graph::operation::Node
{
public:
virtual void accept(NodeVisitor &&) const override;

public:
Node(const graph::operation::Node::InitParam &init_param);

public:
virtual void setInputs(const operand::IndexSet &indexes) override;
virtual void setOutputs(const operand::IndexSet &indexes) override;
};

} // namespace Tanh
} // namespace operation
} // namespace graph
} // namespace neurun

#endif // __NEURUN_GRAPH_OPERATION_TANH_H__

0 comments on commit a560271

Please sign in to comment.