Commit cad5a6b7 authored by Ewa21's avatar Ewa21

[Py] Code formatting

parent 9a9f20cd
...@@ -23,10 +23,10 @@ from ngraph.impl import AxisSet, AxisVector, Coordinate, CoordinateDiff, Functio ...@@ -23,10 +23,10 @@ from ngraph.impl import AxisSet, AxisVector, Coordinate, CoordinateDiff, Functio
from ngraph.impl.op import Abs, Acos, Add, And, Asin, ArgMax, ArgMin, Atan, AvgPool, \ from ngraph.impl.op import Abs, Acos, Add, And, Asin, ArgMax, ArgMin, Atan, AvgPool, \
BatchNormTraining, BatchNormInference, Broadcast, Ceiling, Clamp, Concat, Constant, Convert, \ BatchNormTraining, BatchNormInference, Broadcast, Ceiling, Clamp, Concat, Constant, Convert, \
Convolution, ConvolutionBackpropData, Cos, Cosh, DepthToSpace, Divide, Dot, Elu, Equal, Exp, \ Convolution, ConvolutionBackpropData, Cos, Cosh, DepthToSpace, Divide, Dot, Elu, Equal, Exp, \
Floor, Gelu, Gemm, GetOutputElement, Greater, GreaterEq, GRN, Less, LessEq, Log, LRN, Max, Maximum, \ Floor, Gelu, Gemm, GetOutputElement, Greater, GreaterEq, GRN, Less, LessEq, Log, LRN, Max, \
MaxPool, Min, Minimum, Multiply, Negative, Not, NotEqual, OneHot, Or, Pad, Parameter, Product, \ Maximum, MaxPool, Min, Minimum, Multiply, Negative, Not, NotEqual, OneHot, Or, Pad, \
Power, Relu, ReplaceSlice, Reshape, Reverse, Select, Sign, Sin, Sinh, Slice, Softmax, \ Parameter, Product, Power, Relu, ReplaceSlice, Reshape, Reverse, Select, Sign, Sin, Sinh, \
Sqrt, Subtract, Sum, Tan, Tanh, TopK Slice, Softmax, Sqrt, Subtract, Sum, Tan, Tanh, TopK
from typing import Callable, Iterable, List, Union from typing import Callable, Iterable, List, Union
...@@ -604,7 +604,7 @@ def depth_to_space(node, block_size, name=None): # type: (Node, int, str) -> No ...@@ -604,7 +604,7 @@ def depth_to_space(node, block_size, name=None): # type: (Node, int, str) -> No
""" """
return DepthToSpace(node, block_size) return DepthToSpace(node, block_size)
def gelu(node, name=None): # type: (NodeInput, str) -> Node def gelu(node, name=None): # type: (NodeInput, str) -> Node
r"""Perform Gaussian Error Linear Unit operation element-wise on data from input node. r"""Perform Gaussian Error Linear Unit operation element-wise on data from input node.
...@@ -620,7 +620,7 @@ def gelu(node, name=None): # type: (NodeInput, str) -> Node ...@@ -620,7 +620,7 @@ def gelu(node, name=None): # type: (NodeInput, str) -> Node
:return: The new node performing a GELU operation on its input data element-wise. :return: The new node performing a GELU operation on its input data element-wise.
""" """
return Gelu(as_node(node)) return Gelu(as_node(node))
@nameable_op @nameable_op
def select(selection_node, input_node1, input_node2, name=None): def select(selection_node, input_node1, input_node2, name=None):
......
...@@ -93,6 +93,7 @@ def test_depth_to_space(): ...@@ -93,6 +93,7 @@ def test_depth_to_space():
[12, 18, 13, 19, 14, 20], [12, 18, 13, 19, 14, 20],
[3, 9, 4, 10, 5, 11], [3, 9, 4, 10, 5, 11],
[15, 21, 16, 22, 17, 23]]]], dtype=np.float32) [15, 21, 16, 22, 17, 23]]]], dtype=np.float32)
assert np.allclose(result, expected)
def test_gemm_operator(): def test_gemm_operator():
......
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