Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
N
ngraph
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ngraph
Commits
cad5a6b7
Commit
cad5a6b7
authored
Aug 05, 2019
by
Ewa21
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[Py] Code formatting
parent
9a9f20cd
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
6 deletions
+7
-6
ops.py
python/ngraph/ops.py
+6
-6
test_ops_fused.py
python/test/ngraph/test_ops_fused.py
+1
-0
No files found.
python/ngraph/ops.py
View file @
cad5a6b7
...
@@ -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
,
\
Max
Pool
,
Min
,
Minimum
,
Multiply
,
Negative
,
Not
,
NotEqual
,
OneHot
,
Or
,
Pad
,
Parameter
,
Product
,
\
Max
imum
,
MaxPool
,
Min
,
Minimum
,
Multiply
,
Negative
,
Not
,
NotEqual
,
OneHot
,
Or
,
Pad
,
\
P
ower
,
Relu
,
ReplaceSlice
,
Reshape
,
Reverse
,
Select
,
Sign
,
Sin
,
Sinh
,
Slice
,
Softmax
,
\
P
arameter
,
Product
,
Power
,
Relu
,
ReplaceSlice
,
Reshape
,
Reverse
,
Select
,
Sign
,
Sin
,
Sinh
,
\
Sqrt
,
Subtract
,
Sum
,
Tan
,
Tanh
,
TopK
S
lice
,
Softmax
,
S
qrt
,
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
):
...
...
python/test/ngraph/test_ops_fused.py
View file @
cad5a6b7
...
@@ -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
():
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment