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 @@
* limitations under the License.
*******************************************************************************/
#include <numeric>
#include "ngraph/op/reshape.hpp"
#include "utils/reshape.hpp"
......
......@@ -25,10 +25,10 @@ namespace ngraph
/**
* @brief Permute axes according to specified axes_order parameter.
*
* @param node: The node which axes we want to permute.
* @param axes_order: The permutation of node tensor axes.
* @param node The node which axes we want to permute.
* @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::vector<int> axes_order);
......@@ -36,7 +36,7 @@ namespace ngraph
/**
* @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.
*/
......
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