Commit 2655f5e0 authored by Michał Karzyński's avatar Michał Karzyński Committed by Robert Kimball

[ONNX] Fix reshape helper (#1490)

parent 77fb5b4e
...@@ -14,6 +14,8 @@ ...@@ -14,6 +14,8 @@
* limitations under the License. * limitations under the License.
*******************************************************************************/ *******************************************************************************/
#include <numeric>
#include "ngraph/op/reshape.hpp" #include "ngraph/op/reshape.hpp"
#include "utils/reshape.hpp" #include "utils/reshape.hpp"
......
...@@ -25,10 +25,10 @@ namespace ngraph ...@@ -25,10 +25,10 @@ namespace ngraph
/** /**
* @brief Permute axes according to specified axes_order parameter. * @brief Permute axes according to specified axes_order parameter.
* *
* @param node: The node which axes we want to permute. * @param node The node which axes we want to permute.
* @param axes_order: The permutation of node tensor axes. * @param axes_order The permutation of node tensor axes.
* *
* @retur: New node with permuted axes. * @return: New node with permuted axes.
*/ */
std::shared_ptr<ngraph::Node> reorder_axes(const std::shared_ptr<ngraph::Node>& node, std::shared_ptr<ngraph::Node> reorder_axes(const std::shared_ptr<ngraph::Node>& node,
std::vector<int> axes_order); std::vector<int> axes_order);
...@@ -36,7 +36,7 @@ namespace ngraph ...@@ -36,7 +36,7 @@ namespace ngraph
/** /**
* @brief Return transposed tensor (with axes in reversed order). * @brief Return transposed tensor (with axes in reversed order).
* *
* @param node: Input tensor we want to transpose * @param node Input tensor we want to transpose
* *
* @return: New node with reversed dimensions. * @return: New node with reversed dimensions.
*/ */
......
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