Commit b4104ebe authored by Scott Cyphers's avatar Scott Cyphers

Add concatenate

parent 807ad951
......@@ -18,6 +18,7 @@ set (SRC
tree.cpp
util.cpp
log.cpp
ops/concatenate.cpp
ops/function.cpp
ops/literal.cpp
ops/op.cpp
......
......@@ -71,4 +71,4 @@ namespace ngraph
using UInt8ScalarOp = ScalarLiteralOp<element::UInt8>;
using UInt32ScalarOp = ScalarLiteralOp<element::UInt32>;
using UInt64ScalarOp = ScalarLiteralOp<element::UInt64>;
}
\ No newline at end of file
}
......@@ -24,6 +24,7 @@
#include "ngraph/literal.hpp"
#include "ngraph/node.hpp"
#include "ngraph/op.hpp"
#include "ngraph/ops/concatenate.hpp"
#include "ngraph/parameter.hpp"
#include "ngraph/shape.hpp"
#include "ngraph/type.hpp"
......@@ -32,7 +32,7 @@ namespace ngraph
//Node::ptr candidate();
Node::ptr ceiling(const Node::ptr& arg0, const Node::ptr& arg1);
//Node::ptr concatenate();
//Node::ptr constant();
//Node::ptr convert();
//Node::ptr convolution();
......
// ----------------------------------------------------------------------------
// Copyright 2017 Nervana Systems Inc.
// 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
// ----------------------------------------------------------------------------
#pragma once
namespace ngraph
{
namespace op
{
Node::ptr concatenate(const std::vector<Node::ptr>& args);
}
class ConcatenateOp : public BuiltinOp
{
public:
ConcatenateOp(const std::vector<Node::ptr>& args)
: BuiltinOp(args)
{
}
virtual std::string op_name() const override { return "concatenate"; }
virtual void propagate_types() override;
};
}
// ----------------------------------------------------------------------------
// Copyright 2017 Nervana Systems Inc.
// 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
// ----------------------------------------------------------------------------
#include <memory>
#include "ngraph/ngraph.hpp"
using namespace std;
using namespace ngraph;
void ConcatenateOp::propagate_types()
{
throw ngraph_error("NIY");
}
Node::ptr op::concatenate(const std::vector<Node::ptr>& args)
{
return make_shared<ConcatenateOp>(args);
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment