Commit 9664b96f authored by mbencer's avatar mbencer

Code review remakrs introduced

parent 310fcf07
...@@ -104,7 +104,7 @@ NodeVector builder::opset1::split(const Output<Node>& value, ...@@ -104,7 +104,7 @@ NodeVector builder::opset1::split(const Output<Node>& value,
const std::vector<size_t>& split_lengths, const std::vector<size_t>& split_lengths,
int64_t axis) int64_t axis)
{ {
const auto axis_node = op::Constant::create(element::u64, Shape{}, {axis}); const auto axis_node = ngraph::opset1::Constant::create(element::u64, Shape{}, {axis});
const auto split_lengths_node = const auto split_lengths_node =
ngraph::opset1::Constant::create(element::u64, Shape{split_lengths.size()}, split_lengths); ngraph::opset1::Constant::create(element::u64, Shape{split_lengths.size()}, split_lengths);
const auto variadic_split = const auto variadic_split =
......
...@@ -25,9 +25,9 @@ namespace ngraph ...@@ -25,9 +25,9 @@ namespace ngraph
{ {
/// \brief Split value on specified axis into multiple parts. /// \brief Split value on specified axis into multiple parts.
/// ///
/// \param[in] value The value to be split. /// \param value The value to be split.
/// \param[in] length_parts The vector defining the lengths of each split part. /// \param length_parts The vector defining the lengths of each split part.
/// \param[in] axis The axis we split input node on. Default value is zero axis. /// \param axis The axis we split input node on. Default value is zero axis.
/// ///
/// \return The vector containing multiple nodes we split input node into. /// \return The vector containing multiple nodes we split input node into.
/// ///
...@@ -37,11 +37,11 @@ namespace ngraph ...@@ -37,11 +37,11 @@ namespace ngraph
/// \brief Split node on specified axis into multiple parts. /// \brief Split node on specified axis into multiple parts.
/// ///
/// \param[in] value The value to split. /// \param value The value to split.
/// \param[in] split_parts The number of parts we want to split output at given /// \param split_parts The number of parts we want to split output at given
/// axis. The length of the axis to split must be divisible by /// axis. The length of the axis to split must be divisible by
/// this value. /// this value.
/// \param[in] axis The axis we split input node on. Default value is zero axis. /// \param axis The axis we split input node on. Default value is zero axis.
/// ///
/// \note This implementation supports negative `axis` values (similar to NumPy /// \note This implementation supports negative `axis` values (similar to NumPy
/// indexing). This means that the axis to split on will be counted from /// indexing). This means that the axis to split on will be counted from
...@@ -55,9 +55,9 @@ namespace ngraph ...@@ -55,9 +55,9 @@ namespace ngraph
{ {
/// \brief Split value on specified axis into multiple parts. /// \brief Split value on specified axis into multiple parts.
/// ///
/// \param[in] value The value to be split. /// \param value The value to be split.
/// \param[in] split_lengths The vector defining the lengths of each split part. /// \param split_lengths The vector defining the lengths of each split part.
/// \param[in] axis The axis we split input node on. Default value is zero /// \param axis The axis we split input node on. Default value is zero
/// axis. /// axis.
/// \note This implementation supports negative `axis` values (similar to NumPy /// \note This implementation supports negative `axis` values (similar to NumPy
/// indexing). This means that the axis to split on will be counted from /// indexing). This means that the axis to split on will be counted from
...@@ -70,13 +70,13 @@ namespace ngraph ...@@ -70,13 +70,13 @@ namespace ngraph
const std::vector<size_t>& split_lengths, const std::vector<size_t>& split_lengths,
int64_t axis = 0); int64_t axis = 0);
/// \brief Split node on specified axis into multiple parts. /// \brief Split value on specified axis into multiple parts.
/// ///
/// \param[in] value The value to split. /// \param value The value to split.
/// \param[in] num_splits The number of parts we want to split output at given /// \param num_splits The number of parts we want to split output at given
/// axis. The length of the axis to split must be divisible by /// axis. The length of the axis to split must be divisible by
/// this value. /// this value.
/// \param[in] axis The axis we split input node on. Default value is zero /// \param axis The axis we split input node on. Default value is zero
/// axis. /// axis.
/// ///
/// \note This implementation supports negative `axis` values (similar to NumPy /// \note This implementation supports negative `axis` values (similar to NumPy
......
...@@ -78,7 +78,7 @@ namespace ngraph ...@@ -78,7 +78,7 @@ namespace ngraph
/// ///
/// \brief Factory class which generates sub-graphs for ONNX 'local' pooling /// \brief Factory class which generates sub-graphs for ONNX 'local' pooling
/// operators. /// operators.
/// \note Kernel shape attribute is required /// \note For a 'local' pooling operation, the kernel shape attribute is required
class LocalPoolingFactory : public PoolingFactory class LocalPoolingFactory : public PoolingFactory
{ {
public: public:
...@@ -89,7 +89,8 @@ namespace ngraph ...@@ -89,7 +89,8 @@ namespace ngraph
/// ///
/// \brief Factory class which generates sub-graphs for ONNX 'global' pooling /// \brief Factory class which generates sub-graphs for ONNX 'global' pooling
/// operators. /// operators.
/// \note Kernel shape is calculated based on spatial dims /// \note In a 'global' pooling operation, the kernel shape is calculated
/// based on spatial dims
class GlobalPoolingFactory : public PoolingFactory class GlobalPoolingFactory : public PoolingFactory
{ {
public: public:
......
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