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
410d6e61
Unverified
Commit
410d6e61
authored
5 years ago
by
Scott Cyphers
Committed by
GitHub
5 years ago
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #3368 from NervanaSystems/cyphers/fuseddesc
Convert fused ops to description format
parents
f50e12a1
7f57d4e1
Show whitespace changes
Inline
Side-by-side
Showing
46 changed files
with
154 additions
and
32 deletions
+154
-32
conv_fused.cpp
src/ngraph/op/fused/conv_fused.cpp
+8
-5
conv_fused.hpp
src/ngraph/op/fused/conv_fused.hpp
+9
-0
depth_to_space.cpp
src/ngraph/op/fused/depth_to_space.cpp
+3
-1
depth_to_space.hpp
src/ngraph/op/fused/depth_to_space.hpp
+3
-0
elu.cpp
src/ngraph/op/fused/elu.cpp
+3
-1
elu.hpp
src/ngraph/op/fused/elu.hpp
+3
-0
fake_quantize.cpp
src/ngraph/op/fused/fake_quantize.cpp
+3
-1
fake_quantize.hpp
src/ngraph/op/fused/fake_quantize.hpp
+3
-0
gelu.cpp
src/ngraph/op/fused/gelu.cpp
+3
-1
gelu.hpp
src/ngraph/op/fused/gelu.hpp
+3
-0
gemm.cpp
src/ngraph/op/fused/gemm.cpp
+3
-1
gemm.hpp
src/ngraph/op/fused/gemm.hpp
+3
-0
grn.cpp
src/ngraph/op/fused/grn.cpp
+3
-1
grn.hpp
src/ngraph/op/fused/grn.hpp
+3
-0
group_conv_transpose.cpp
src/ngraph/op/fused/group_conv_transpose.cpp
+3
-1
group_conv_transpose.hpp
src/ngraph/op/fused/group_conv_transpose.hpp
+3
-0
gru_cell.cpp
src/ngraph/op/fused/gru_cell.cpp
+4
-2
gru_cell.hpp
src/ngraph/op/fused/gru_cell.hpp
+3
-0
hard_sigmoid.cpp
src/ngraph/op/fused/hard_sigmoid.cpp
+3
-1
hard_sigmoid.hpp
src/ngraph/op/fused/hard_sigmoid.hpp
+3
-0
leaky_relu.cpp
src/ngraph/op/fused/leaky_relu.cpp
+3
-1
leaky_relu.hpp
src/ngraph/op/fused/leaky_relu.hpp
+3
-0
lstm_cell.cpp
src/ngraph/op/fused/lstm_cell.cpp
+4
-2
lstm_cell.hpp
src/ngraph/op/fused/lstm_cell.hpp
+3
-0
mvn.cpp
src/ngraph/op/fused/mvn.cpp
+4
-2
mvn.hpp
src/ngraph/op/fused/mvn.hpp
+3
-0
normalize.cpp
src/ngraph/op/fused/normalize.cpp
+3
-1
normalize.hpp
src/ngraph/op/fused/normalize.hpp
+3
-0
prelu.cpp
src/ngraph/op/fused/prelu.cpp
+3
-1
prelu.hpp
src/ngraph/op/fused/prelu.hpp
+3
-0
rnn_cell.cpp
src/ngraph/op/fused/rnn_cell.cpp
+4
-2
rnn_cell.hpp
src/ngraph/op/fused/rnn_cell.hpp
+3
-0
scale_shift.cpp
src/ngraph/op/fused/scale_shift.cpp
+3
-1
scale_shift.hpp
src/ngraph/op/fused/scale_shift.hpp
+3
-0
shuffle_channels.cpp
src/ngraph/op/fused/shuffle_channels.cpp
+3
-1
shuffle_channels.hpp
src/ngraph/op/fused/shuffle_channels.hpp
+3
-0
space_to_depth.cpp
src/ngraph/op/fused/space_to_depth.cpp
+3
-1
space_to_depth.hpp
src/ngraph/op/fused/space_to_depth.hpp
+3
-0
split.cpp
src/ngraph/op/fused/split.cpp
+4
-2
split.hpp
src/ngraph/op/fused/split.hpp
+3
-0
squared_difference.cpp
src/ngraph/op/fused/squared_difference.cpp
+3
-1
squared_difference.hpp
src/ngraph/op/fused/squared_difference.hpp
+3
-0
squeeze.cpp
src/ngraph/op/fused/squeeze.cpp
+3
-1
squeeze.hpp
src/ngraph/op/fused/squeeze.hpp
+3
-0
unsqueeze.cpp
src/ngraph/op/fused/unsqueeze.cpp
+3
-1
unsqueeze.hpp
src/ngraph/op/fused/unsqueeze.hpp
+3
-0
No files found.
src/ngraph/op/fused/conv_fused.cpp
View file @
410d6e61
...
@@ -30,6 +30,11 @@
...
@@ -30,6 +30,11 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
ConvolutionBias
::
type_name
{
"ConvolutionBias"
};
const
string
op
::
ConvolutionBiasBackpropFiltersBias
::
type_name
{
"ConvolutionBiasBackpropFiltersBias"
};
const
string
op
::
ConvolutionBiasAdd
::
type_name
{
"ConvolutionBiasAdd"
};
static
void
validate_convbias_shapes
(
const
Node
*
node
,
static
void
validate_convbias_shapes
(
const
Node
*
node
,
element
::
Type
et_filters
,
element
::
Type
et_filters
,
element
::
Type
et_bias
,
element
::
Type
et_bias
,
...
@@ -75,7 +80,7 @@ op::ConvolutionBias::ConvolutionBias(const shared_ptr<Node>& data_batch,
...
@@ -75,7 +80,7 @@ op::ConvolutionBias::ConvolutionBias(const shared_ptr<Node>& data_batch,
const
CoordinateDiff
&
padding_above
,
const
CoordinateDiff
&
padding_above
,
const
Strides
&
data_dilation_strides
,
const
Strides
&
data_dilation_strides
,
const
bool
with_relu
)
const
bool
with_relu
)
:
FusedOp
(
"ConvolutionBias"
,
check_single_output_args
({
data_batch
,
filters
,
bias
}))
:
FusedOp
(
check_single_output_args
({
data_batch
,
filters
,
bias
}))
,
m_window_movement_strides
(
window_movement_strides
)
,
m_window_movement_strides
(
window_movement_strides
)
,
m_window_dilation_strides
(
window_dilation_strides
)
,
m_window_dilation_strides
(
window_dilation_strides
)
,
m_padding_below
(
padding_below
)
,
m_padding_below
(
padding_below
)
...
@@ -280,8 +285,7 @@ op::ConvolutionBiasBackpropFiltersBias::ConvolutionBiasBackpropFiltersBias(
...
@@ -280,8 +285,7 @@ op::ConvolutionBiasBackpropFiltersBias::ConvolutionBiasBackpropFiltersBias(
const
CoordinateDiff
&
padding_below_forward
,
const
CoordinateDiff
&
padding_below_forward
,
const
CoordinateDiff
&
padding_above_forward
,
const
CoordinateDiff
&
padding_above_forward
,
const
Strides
&
data_dilation_strides_forward
)
const
Strides
&
data_dilation_strides_forward
)
:
FusedOp
(
"ConvolutionBiasBackpropFiltersBias"
,
:
FusedOp
(
check_single_output_args
({
data_batch
,
output_delta
}))
check_single_output_args
({
data_batch
,
output_delta
}))
,
m_filters_shape
(
filters_shape
)
,
m_filters_shape
(
filters_shape
)
,
m_bias_shape
(
bias_shape
)
,
m_bias_shape
(
bias_shape
)
,
m_window_movement_strides_forward
(
window_movement_strides_forward
)
,
m_window_movement_strides_forward
(
window_movement_strides_forward
)
...
@@ -368,8 +372,7 @@ op::ConvolutionBiasAdd::ConvolutionBiasAdd(const std::shared_ptr<Node>& data_bat
...
@@ -368,8 +372,7 @@ op::ConvolutionBiasAdd::ConvolutionBiasAdd(const std::shared_ptr<Node>& data_bat
const
CoordinateDiff
&
padding_above
,
const
CoordinateDiff
&
padding_above
,
const
Strides
&
data_dilation_strides
,
const
Strides
&
data_dilation_strides
,
bool
with_relu
)
bool
with_relu
)
:
FusedOp
(
"ConvolutionBiasAdd"
,
:
FusedOp
(
check_single_output_args
({
data_batch
,
filters
,
bias
,
add_input
}))
check_single_output_args
({
data_batch
,
filters
,
bias
,
add_input
}))
,
m_window_movement_strides
(
window_movement_strides
)
,
m_window_movement_strides
(
window_movement_strides
)
,
m_window_dilation_strides
(
window_dilation_strides
)
,
m_window_dilation_strides
(
window_dilation_strides
)
,
m_padding_below
(
padding_below
)
,
m_padding_below
(
padding_below
)
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/conv_fused.hpp
View file @
410d6e61
...
@@ -28,6 +28,9 @@ namespace ngraph
...
@@ -28,6 +28,9 @@ namespace ngraph
class
ConvolutionBias
:
public
ngraph
::
op
::
util
::
FusedOp
class
ConvolutionBias
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
ConvolutionBias
(
const
std
::
shared_ptr
<
op
::
Convolution
>&
conv
,
ConvolutionBias
(
const
std
::
shared_ptr
<
op
::
Convolution
>&
conv
,
const
std
::
shared_ptr
<
Node
>&
bias
,
const
std
::
shared_ptr
<
Node
>&
bias
,
const
bool
with_relu
=
false
);
const
bool
with_relu
=
false
);
...
@@ -79,6 +82,9 @@ namespace ngraph
...
@@ -79,6 +82,9 @@ namespace ngraph
class
ConvolutionBiasBackpropFiltersBias
:
public
ngraph
::
op
::
util
::
FusedOp
class
ConvolutionBiasBackpropFiltersBias
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
ConvolutionBiasBackpropFiltersBias
(
const
std
::
shared_ptr
<
Node
>&
data_batch
,
ConvolutionBiasBackpropFiltersBias
(
const
std
::
shared_ptr
<
Node
>&
data_batch
,
const
Shape
&
filters_shape
,
const
Shape
&
filters_shape
,
const
Shape
&
bias_shape
,
const
Shape
&
bias_shape
,
...
@@ -169,6 +175,9 @@ namespace ngraph
...
@@ -169,6 +175,9 @@ namespace ngraph
class
ConvolutionBiasAdd
:
public
ngraph
::
op
::
util
::
FusedOp
class
ConvolutionBiasAdd
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
ConvolutionBiasAdd
(
const
std
::
shared_ptr
<
op
::
ConvolutionBias
>&
conv
,
ConvolutionBiasAdd
(
const
std
::
shared_ptr
<
op
::
ConvolutionBias
>&
conv
,
const
std
::
shared_ptr
<
Node
>&
sum_input
,
const
std
::
shared_ptr
<
Node
>&
sum_input
,
bool
with_relu
=
false
);
bool
with_relu
=
false
);
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/depth_to_space.cpp
View file @
410d6e61
...
@@ -25,8 +25,10 @@
...
@@ -25,8 +25,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
DepthToSpace
::
type_name
{
"DepthToSpace"
};
op
::
DepthToSpace
::
DepthToSpace
(
const
shared_ptr
<
Node
>&
data
,
const
size_t
block_size
)
op
::
DepthToSpace
::
DepthToSpace
(
const
shared_ptr
<
Node
>&
data
,
const
size_t
block_size
)
:
FusedOp
(
"DepthToSpace"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_blocksize
(
block_size
)
,
m_blocksize
(
block_size
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/depth_to_space.hpp
View file @
410d6e61
...
@@ -34,6 +34,9 @@ namespace ngraph
...
@@ -34,6 +34,9 @@ namespace ngraph
class
DepthToSpace
:
public
ngraph
::
op
::
util
::
FusedOp
class
DepthToSpace
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs a DepthToSpace operation.
/// \brief Constructs a DepthToSpace operation.
///
///
/// \param data - Node producing the input tensor
/// \param data - Node producing the input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/elu.cpp
View file @
410d6e61
...
@@ -28,8 +28,10 @@
...
@@ -28,8 +28,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
Elu
::
type_name
{
"Elu"
};
op
::
Elu
::
Elu
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
alpha
)
op
::
Elu
::
Elu
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
alpha
)
:
FusedOp
(
"Elu"
,
{
data
,
alpha
}
)
:
FusedOp
(
check_single_output_args
({
data
,
alpha
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/elu.hpp
View file @
410d6e61
...
@@ -31,6 +31,9 @@ namespace ngraph
...
@@ -31,6 +31,9 @@ namespace ngraph
class
Elu
:
public
ngraph
::
op
::
util
::
FusedOp
class
Elu
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs an Elu operation.
/// \brief Constructs an Elu operation.
///
///
/// \param data Input tensor
/// \param data Input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/fake_quantize.cpp
View file @
410d6e61
...
@@ -36,13 +36,15 @@
...
@@ -36,13 +36,15 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
FakeQuantize
::
type_name
{
"FakeQuantize"
};
op
::
FakeQuantize
::
FakeQuantize
(
const
shared_ptr
<
Node
>&
data
,
op
::
FakeQuantize
::
FakeQuantize
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
input_low
,
const
shared_ptr
<
Node
>&
input_low
,
const
shared_ptr
<
Node
>&
input_high
,
const
shared_ptr
<
Node
>&
input_high
,
const
shared_ptr
<
Node
>&
output_low
,
const
shared_ptr
<
Node
>&
output_low
,
const
shared_ptr
<
Node
>&
output_high
,
const
shared_ptr
<
Node
>&
output_high
,
size_t
levels
)
size_t
levels
)
:
FusedOp
(
"FakeQuantize"
,
{
data
,
input_low
,
input_high
,
output_low
,
output_high
}
)
:
FusedOp
(
check_single_output_args
({
data
,
input_low
,
input_high
,
output_low
,
output_high
})
)
,
m_levels
(
levels
)
,
m_levels
(
levels
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/fake_quantize.hpp
View file @
410d6e61
...
@@ -38,6 +38,9 @@ namespace ngraph
...
@@ -38,6 +38,9 @@ namespace ngraph
class
FakeQuantize
:
public
ngraph
::
op
::
util
::
FusedOp
class
FakeQuantize
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
///
///
/// \brief Constructs a FakeQuantize operation node.
/// \brief Constructs a FakeQuantize operation node.
///
///
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/gelu.cpp
View file @
410d6e61
...
@@ -27,8 +27,10 @@
...
@@ -27,8 +27,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
Gelu
::
type_name
{
"Gelu"
};
op
::
Gelu
::
Gelu
(
const
shared_ptr
<
Node
>&
data
)
op
::
Gelu
::
Gelu
(
const
shared_ptr
<
Node
>&
data
)
:
FusedOp
(
"Gelu"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/gelu.hpp
View file @
410d6e61
...
@@ -32,6 +32,9 @@ namespace ngraph
...
@@ -32,6 +32,9 @@ namespace ngraph
class
Gelu
:
public
ngraph
::
op
::
util
::
FusedOp
class
Gelu
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs an Gelu operation.
/// \brief Constructs an Gelu operation.
///
///
/// \param data Input tensor
/// \param data Input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/gemm.cpp
View file @
410d6e61
...
@@ -25,6 +25,8 @@
...
@@ -25,6 +25,8 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
Gemm
::
type_name
{
"Gemm"
};
op
::
Gemm
::
Gemm
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
A
,
op
::
Gemm
::
Gemm
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
A
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
B
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
B
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
C
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
C
,
...
@@ -32,7 +34,7 @@ op::Gemm::Gemm(const std::shared_ptr<ngraph::Node>& A,
...
@@ -32,7 +34,7 @@ op::Gemm::Gemm(const std::shared_ptr<ngraph::Node>& A,
double
beta
,
double
beta
,
bool
transA
,
bool
transA
,
bool
transB
)
bool
transB
)
:
FusedOp
(
"Gemm"
,
{
A
,
B
,
C
}
)
:
FusedOp
(
check_single_output_args
({
A
,
B
,
C
})
)
,
m_alpha
{
alpha
}
,
m_alpha
{
alpha
}
,
m_beta
{
beta
}
,
m_beta
{
beta
}
,
m_transA
{
transA
}
,
m_transA
{
transA
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/gemm.hpp
View file @
410d6e61
...
@@ -36,6 +36,9 @@ namespace ngraph
...
@@ -36,6 +36,9 @@ namespace ngraph
class
Gemm
:
public
ngraph
::
op
::
util
::
FusedOp
class
Gemm
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs an Gemm operation.
/// \brief Constructs an Gemm operation.
///
///
/// \param A Input tensor A
/// \param A Input tensor A
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/grn.cpp
View file @
410d6e61
...
@@ -27,8 +27,10 @@
...
@@ -27,8 +27,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
GRN
::
type_name
{
"GRN"
};
op
::
GRN
::
GRN
(
const
shared_ptr
<
Node
>&
data
,
float
bias
)
op
::
GRN
::
GRN
(
const
shared_ptr
<
Node
>&
data
,
float
bias
)
:
FusedOp
(
"GRN"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_bias
(
bias
)
,
m_bias
(
bias
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/grn.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
GRN
:
public
ngraph
::
op
::
util
::
FusedOp
class
GRN
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs a GRN operation.
/// \brief Constructs a GRN operation.
///
///
/// \param data - Node producing the input tensor
/// \param data - Node producing the input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/group_conv_transpose.cpp
View file @
410d6e61
...
@@ -27,6 +27,8 @@
...
@@ -27,6 +27,8 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
GroupConvolutionTranspose
::
type_name
{
"GroupConvolutionTranspose"
};
op
::
GroupConvolutionTranspose
::
GroupConvolutionTranspose
(
const
shared_ptr
<
Node
>&
data
,
op
::
GroupConvolutionTranspose
::
GroupConvolutionTranspose
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
filters
,
const
shared_ptr
<
Node
>&
filters
,
const
Strides
&
strides
,
const
Strides
&
strides
,
...
@@ -37,7 +39,7 @@ op::GroupConvolutionTranspose::GroupConvolutionTranspose(const shared_ptr<Node>&
...
@@ -37,7 +39,7 @@ op::GroupConvolutionTranspose::GroupConvolutionTranspose(const shared_ptr<Node>&
const
size_t
groups
,
const
size_t
groups
,
const
PadType
&
pad_type
,
const
PadType
&
pad_type
,
const
Shape
&
output_shape
)
const
Shape
&
output_shape
)
:
FusedOp
(
"GroupConvolutionTranspose"
,
check_single_output_args
({
data
,
filters
}))
:
FusedOp
(
check_single_output_args
({
data
,
filters
}))
,
m_strides
(
strides
)
,
m_strides
(
strides
)
,
m_dilations
(
dilations
)
,
m_dilations
(
dilations
)
,
m_padding_begin
(
padding_begin
)
,
m_padding_begin
(
padding_begin
)
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/group_conv_transpose.hpp
View file @
410d6e61
...
@@ -35,6 +35,9 @@ namespace ngraph
...
@@ -35,6 +35,9 @@ namespace ngraph
class
GroupConvolutionTranspose
:
public
util
::
FusedOp
class
GroupConvolutionTranspose
:
public
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
///
///
/// \brief Constructs GroupConvolutionTranspose operation.
/// \brief Constructs GroupConvolutionTranspose operation.
///
///
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/gru_cell.cpp
View file @
410d6e61
...
@@ -31,6 +31,8 @@
...
@@ -31,6 +31,8 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
GRUCell
::
type_name
{
"GRUCell"
};
op
::
GRUCell
::
GRUCell
(
const
shared_ptr
<
Node
>&
X
,
op
::
GRUCell
::
GRUCell
(
const
shared_ptr
<
Node
>&
X
,
const
shared_ptr
<
Node
>&
W
,
const
shared_ptr
<
Node
>&
W
,
const
shared_ptr
<
Node
>&
R
,
const
shared_ptr
<
Node
>&
R
,
...
@@ -59,7 +61,7 @@ op::GRUCell::GRUCell(const shared_ptr<Node>& X,
...
@@ -59,7 +61,7 @@ op::GRUCell::GRUCell(const shared_ptr<Node>& X,
const
vector
<
float
>&
activation_beta
,
const
vector
<
float
>&
activation_beta
,
float
clip
,
float
clip
,
bool
linear_before_reset
)
bool
linear_before_reset
)
:
FusedOp
(
"GRUCell"
,
{
X
,
W
,
R
,
H_t
}
)
:
FusedOp
(
check_single_output_args
({
X
,
W
,
R
,
H_t
})
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_g
{
get_activation_function
(
1
)}
,
m_activation_g
{
get_activation_function
(
1
)}
...
@@ -80,7 +82,7 @@ op::GRUCell::GRUCell(const shared_ptr<Node>& X,
...
@@ -80,7 +82,7 @@ op::GRUCell::GRUCell(const shared_ptr<Node>& X,
const
vector
<
float
>&
activation_beta
,
const
vector
<
float
>&
activation_beta
,
float
clip
,
float
clip
,
bool
linear_before_reset
)
bool
linear_before_reset
)
:
FusedOp
(
"GRUCell"
,
{
X
,
W
,
R
,
H_t
,
B
}
)
:
FusedOp
(
check_single_output_args
({
X
,
W
,
R
,
H_t
,
B
})
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_g
{
get_activation_function
(
1
)}
,
m_activation_g
{
get_activation_function
(
1
)}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/gru_cell.hpp
View file @
410d6e61
...
@@ -41,6 +41,9 @@ namespace ngraph
...
@@ -41,6 +41,9 @@ namespace ngraph
class
GRUCell
:
public
util
::
FusedOp
,
public
util
::
RNNCellBase
class
GRUCell
:
public
util
::
FusedOp
,
public
util
::
RNNCellBase
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
///
///
/// \brief Constructs GRUCell node.
/// \brief Constructs GRUCell node.
///
///
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/hard_sigmoid.cpp
View file @
410d6e61
...
@@ -27,8 +27,10 @@
...
@@ -27,8 +27,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
HardSigmoid
::
type_name
{
"HardSigmoid"
};
op
::
HardSigmoid
::
HardSigmoid
(
const
shared_ptr
<
Node
>&
data
,
float
alpha
,
float
beta
)
op
::
HardSigmoid
::
HardSigmoid
(
const
shared_ptr
<
Node
>&
data
,
float
alpha
,
float
beta
)
:
FusedOp
(
"HardSigmoid"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_alpha
(
alpha
)
,
m_alpha
(
alpha
)
,
m_beta
(
beta
)
,
m_beta
(
beta
)
{
{
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/hard_sigmoid.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
HardSigmoid
:
public
ngraph
::
op
::
util
::
FusedOp
class
HardSigmoid
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs a HardSigmoid operation.
/// \brief Constructs a HardSigmoid operation.
///
///
/// \param data Input tensor.
/// \param data Input tensor.
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/leaky_relu.cpp
View file @
410d6e61
...
@@ -23,8 +23,10 @@
...
@@ -23,8 +23,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
LeakyRelu
::
type_name
{
"LeakyRelu"
};
op
::
LeakyRelu
::
LeakyRelu
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
alpha
)
op
::
LeakyRelu
::
LeakyRelu
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
alpha
)
:
FusedOp
(
"LeakyRelu"
,
{
data
,
alpha
}
)
:
FusedOp
(
check_single_output_args
({
data
,
alpha
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/leaky_relu.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
LeakyRelu
:
public
ngraph
::
op
::
util
::
FusedOp
class
LeakyRelu
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
LeakyRelu
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
LeakyRelu
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
alpha
);
const
std
::
shared_ptr
<
ngraph
::
Node
>&
alpha
);
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/lstm_cell.cpp
View file @
410d6e61
...
@@ -31,6 +31,8 @@
...
@@ -31,6 +31,8 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
LSTMCell
::
type_name
{
"LSTMCell"
};
op
::
LSTMCell
::
LSTMCell
(
const
shared_ptr
<
Node
>&
X
,
op
::
LSTMCell
::
LSTMCell
(
const
shared_ptr
<
Node
>&
X
,
const
shared_ptr
<
Node
>&
W
,
const
shared_ptr
<
Node
>&
W
,
const
shared_ptr
<
Node
>&
R
,
const
shared_ptr
<
Node
>&
R
,
...
@@ -62,7 +64,7 @@ op::LSTMCell::LSTMCell(const shared_ptr<Node>& X,
...
@@ -62,7 +64,7 @@ op::LSTMCell::LSTMCell(const shared_ptr<Node>& X,
const
vector
<
float
>&
activation_beta
,
const
vector
<
float
>&
activation_beta
,
float
clip
,
float
clip
,
bool
input_forget
)
bool
input_forget
)
:
FusedOp
(
"LSTMCell"
,
{
X
,
W
,
R
,
H_t
,
C_t
}
)
:
FusedOp
(
check_single_output_args
({
X
,
W
,
R
,
H_t
,
C_t
})
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_g
{
get_activation_function
(
1
)}
,
m_activation_g
{
get_activation_function
(
1
)}
...
@@ -87,7 +89,7 @@ op::LSTMCell::LSTMCell(const shared_ptr<Node>& X,
...
@@ -87,7 +89,7 @@ op::LSTMCell::LSTMCell(const shared_ptr<Node>& X,
const
vector
<
float
>&
activation_beta
,
const
vector
<
float
>&
activation_beta
,
float
clip
,
float
clip
,
bool
input_forget
)
bool
input_forget
)
:
FusedOp
(
"LSTMCell"
,
{
X
,
W
,
R
,
H_t
,
C_t
,
B
,
P
}
)
:
FusedOp
(
check_single_output_args
({
X
,
W
,
R
,
H_t
,
C_t
,
B
,
P
})
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_g
{
get_activation_function
(
1
)}
,
m_activation_g
{
get_activation_function
(
1
)}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/lstm_cell.hpp
View file @
410d6e61
...
@@ -42,6 +42,9 @@ namespace ngraph
...
@@ -42,6 +42,9 @@ namespace ngraph
class
LSTMCell
:
public
util
::
FusedOp
,
public
util
::
RNNCellBase
class
LSTMCell
:
public
util
::
FusedOp
,
public
util
::
RNNCellBase
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
///
///
/// \brief Constructs LSTMCell node.
/// \brief Constructs LSTMCell node.
///
///
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/mvn.cpp
View file @
410d6e61
...
@@ -27,11 +27,13 @@
...
@@ -27,11 +27,13 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
MVN
::
type_name
{
"MVN"
};
op
::
MVN
::
MVN
(
const
std
::
shared_ptr
<
Node
>&
data
,
op
::
MVN
::
MVN
(
const
std
::
shared_ptr
<
Node
>&
data
,
bool
across_channels
,
bool
across_channels
,
bool
normalize_variance
,
bool
normalize_variance
,
double
eps
)
double
eps
)
:
FusedOp
(
"MVN"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_eps
{
eps
}
,
m_eps
{
eps
}
,
m_across_channels
{
across_channels
}
,
m_across_channels
{
across_channels
}
,
m_normalize_variance
{
normalize_variance
}
,
m_normalize_variance
{
normalize_variance
}
...
@@ -52,7 +54,7 @@ op::MVN::MVN(const std::shared_ptr<Node>& data,
...
@@ -52,7 +54,7 @@ op::MVN::MVN(const std::shared_ptr<Node>& data,
AxisSet
reduction_axes
,
AxisSet
reduction_axes
,
bool
normalize_variance
,
bool
normalize_variance
,
double
eps
)
double
eps
)
:
FusedOp
(
"MVN"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_eps
{
eps
}
,
m_eps
{
eps
}
,
m_across_channels
{
false
}
,
m_across_channels
{
false
}
,
m_normalize_variance
{
normalize_variance
}
,
m_normalize_variance
{
normalize_variance
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/mvn.hpp
View file @
410d6e61
...
@@ -29,6 +29,9 @@ namespace ngraph
...
@@ -29,6 +29,9 @@ namespace ngraph
class
MVN
:
public
ngraph
::
op
::
util
::
FusedOp
class
MVN
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs an MVN operation.
/// \brief Constructs an MVN operation.
///
///
/// \param data Input tensor with data
/// \param data Input tensor with data
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/normalize.cpp
View file @
410d6e61
...
@@ -26,12 +26,14 @@
...
@@ -26,12 +26,14 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
Normalize
::
type_name
{
"Normalize"
};
op
::
Normalize
::
Normalize
(
const
shared_ptr
<
ngraph
::
Node
>&
data
,
op
::
Normalize
::
Normalize
(
const
shared_ptr
<
ngraph
::
Node
>&
data
,
const
shared_ptr
<
ngraph
::
Node
>&
scale
,
const
shared_ptr
<
ngraph
::
Node
>&
scale
,
bool
across_spatial
,
bool
across_spatial
,
bool
channel_shared
,
bool
channel_shared
,
float
eps
)
float
eps
)
:
FusedOp
(
"Normalize"
,
{
data
,
scale
}
)
:
FusedOp
(
check_single_output_args
({
data
,
scale
})
)
,
m_across_spatial
{
across_spatial
}
,
m_across_spatial
{
across_spatial
}
,
m_channel_shared
{
channel_shared
}
,
m_channel_shared
{
channel_shared
}
,
m_eps
{
eps
}
,
m_eps
{
eps
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/normalize.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
Normalize
:
public
ngraph
::
op
::
util
::
FusedOp
class
Normalize
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
///
///
/// \brief Constructs a Normalize operation.
/// \brief Constructs a Normalize operation.
///
///
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/prelu.cpp
View file @
410d6e61
...
@@ -29,8 +29,10 @@
...
@@ -29,8 +29,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
PRelu
::
type_name
{
"PRelu"
};
op
::
PRelu
::
PRelu
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
slope
)
op
::
PRelu
::
PRelu
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
slope
)
:
FusedOp
(
"PRelu"
,
{
data
,
slope
}
)
:
FusedOp
(
check_single_output_args
({
data
,
slope
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/prelu.hpp
View file @
410d6e61
...
@@ -31,6 +31,9 @@ namespace ngraph
...
@@ -31,6 +31,9 @@ namespace ngraph
class
PRelu
:
public
ngraph
::
op
::
util
::
FusedOp
class
PRelu
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs a PRelu operation.
/// \brief Constructs a PRelu operation.
///
///
/// \param data Input tensor
/// \param data Input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/rnn_cell.cpp
View file @
410d6e61
...
@@ -31,6 +31,8 @@
...
@@ -31,6 +31,8 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
RNNCell
::
type_name
{
"RNNCell"
};
op
::
RNNCell
::
RNNCell
(
const
shared_ptr
<
Node
>&
X
,
op
::
RNNCell
::
RNNCell
(
const
shared_ptr
<
Node
>&
X
,
const
shared_ptr
<
Node
>&
W
,
const
shared_ptr
<
Node
>&
W
,
const
shared_ptr
<
Node
>&
R
,
const
shared_ptr
<
Node
>&
R
,
...
@@ -50,7 +52,7 @@ op::RNNCell::RNNCell(const shared_ptr<Node>& X,
...
@@ -50,7 +52,7 @@ op::RNNCell::RNNCell(const shared_ptr<Node>& X,
const
vector
<
float
>&
activation_alpha
,
const
vector
<
float
>&
activation_alpha
,
const
vector
<
float
>&
activation_beta
,
const
vector
<
float
>&
activation_beta
,
float
clip
)
float
clip
)
:
FusedOp
(
"RNNCell"
,
{
X
,
W
,
R
,
H_t
}
)
:
FusedOp
(
check_single_output_args
({
X
,
W
,
R
,
H_t
})
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_f
{
get_activation_function
(
0
)}
{
{
...
@@ -68,7 +70,7 @@ op::RNNCell::RNNCell(const shared_ptr<Node>& X,
...
@@ -68,7 +70,7 @@ op::RNNCell::RNNCell(const shared_ptr<Node>& X,
const
vector
<
float
>&
activation_alpha
,
const
vector
<
float
>&
activation_alpha
,
const
vector
<
float
>&
activation_beta
,
const
vector
<
float
>&
activation_beta
,
float
clip
)
float
clip
)
:
FusedOp
(
"RNNCell"
,
{
X
,
W
,
R
,
H_t
,
B
}
)
:
FusedOp
(
check_single_output_args
({
X
,
W
,
R
,
H_t
,
B
})
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
RNNCellBase
(
hidden_size
,
clip
,
activations
,
activation_alpha
,
activation_beta
)
,
m_activation_f
{
get_activation_function
(
0
)}
,
m_activation_f
{
get_activation_function
(
0
)}
{
{
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/rnn_cell.hpp
View file @
410d6e61
...
@@ -41,6 +41,9 @@ namespace ngraph
...
@@ -41,6 +41,9 @@ namespace ngraph
class
RNNCell
:
public
util
::
FusedOp
,
public
util
::
RNNCellBase
class
RNNCell
:
public
util
::
FusedOp
,
public
util
::
RNNCellBase
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
///
///
/// \brief Constructs RNNCell node.
/// \brief Constructs RNNCell node.
///
///
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/scale_shift.cpp
View file @
410d6e61
...
@@ -21,10 +21,12 @@
...
@@ -21,10 +21,12 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
ScaleShift
::
type_name
{
"ScaleShift"
};
op
::
ScaleShift
::
ScaleShift
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
op
::
ScaleShift
::
ScaleShift
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
scale
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
scale
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
shift
)
const
std
::
shared_ptr
<
ngraph
::
Node
>&
shift
)
:
FusedOp
(
"ScaleShift"
,
{
data
,
scale
,
shift
}
)
:
FusedOp
(
check_single_output_args
({
data
,
scale
,
shift
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/scale_shift.hpp
View file @
410d6e61
...
@@ -31,6 +31,9 @@ namespace ngraph
...
@@ -31,6 +31,9 @@ namespace ngraph
class
ScaleShift
:
public
ngraph
::
op
::
util
::
FusedOp
class
ScaleShift
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs an ScaleShift operation.
/// \brief Constructs an ScaleShift operation.
///
///
/// \param data Input tensor
/// \param data Input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/shuffle_channels.cpp
View file @
410d6e61
...
@@ -21,10 +21,12 @@
...
@@ -21,10 +21,12 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
ShuffleChannels
::
type_name
{
"ShuffleChannels"
};
op
::
ShuffleChannels
::
ShuffleChannels
(
const
shared_ptr
<
Node
>&
data
,
op
::
ShuffleChannels
::
ShuffleChannels
(
const
shared_ptr
<
Node
>&
data
,
const
int
axis
,
const
int
axis
,
const
size_t
groups
)
const
size_t
groups
)
:
FusedOp
(
"ShuffleChannels"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_axis
(
axis
)
,
m_axis
(
axis
)
,
m_groups
{
groups
}
,
m_groups
{
groups
}
{
{
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/shuffle_channels.hpp
View file @
410d6e61
...
@@ -29,6 +29,9 @@ namespace ngraph
...
@@ -29,6 +29,9 @@ namespace ngraph
class
ShuffleChannels
:
public
ngraph
::
op
::
util
::
FusedOp
class
ShuffleChannels
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs a ShuffleChannels node.
/// \brief Constructs a ShuffleChannels node.
///
///
/// \param data - Node producing the input tensor
/// \param data - Node producing the input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/space_to_depth.cpp
View file @
410d6e61
...
@@ -24,8 +24,10 @@
...
@@ -24,8 +24,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
SpaceToDepth
::
type_name
{
"SpaceToDepth"
};
op
::
SpaceToDepth
::
SpaceToDepth
(
const
shared_ptr
<
Node
>&
data
,
const
size_t
block_size
)
op
::
SpaceToDepth
::
SpaceToDepth
(
const
shared_ptr
<
Node
>&
data
,
const
size_t
block_size
)
:
FusedOp
(
"SpaceToDepth"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_blocksize
(
block_size
)
,
m_blocksize
(
block_size
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/space_to_depth.hpp
View file @
410d6e61
...
@@ -32,6 +32,9 @@ namespace ngraph
...
@@ -32,6 +32,9 @@ namespace ngraph
class
SpaceToDepth
:
public
ngraph
::
op
::
util
::
FusedOp
class
SpaceToDepth
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs a SpaceToDepth operation.
/// \brief Constructs a SpaceToDepth operation.
///
///
/// \param data - Node producing the input tensor
/// \param data - Node producing the input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/split.cpp
View file @
410d6e61
...
@@ -21,8 +21,10 @@
...
@@ -21,8 +21,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
Split
::
type_name
{
"Split"
};
op
::
Split
::
Split
(
const
shared_ptr
<
Node
>&
data
,
const
int
axis
,
const
size_t
num_split
)
op
::
Split
::
Split
(
const
shared_ptr
<
Node
>&
data
,
const
int
axis
,
const
size_t
num_split
)
:
FusedOp
(
"Split"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_split_evenly
{
true
}
,
m_split_evenly
{
true
}
,
m_axis
{
axis
}
,
m_axis
{
axis
}
,
m_num_split
{
num_split
}
,
m_num_split
{
num_split
}
...
@@ -33,7 +35,7 @@ op::Split::Split(const shared_ptr<Node>& data, const int axis, const size_t num_
...
@@ -33,7 +35,7 @@ op::Split::Split(const shared_ptr<Node>& data, const int axis, const size_t num_
op
::
Split
::
Split
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
op
::
Split
::
Split
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
const
int
axis
,
const
int
axis
,
const
std
::
vector
<
size_t
>&
splits
)
const
std
::
vector
<
size_t
>&
splits
)
:
FusedOp
(
"Split"
,
{
data
}
)
:
FusedOp
(
check_single_output_args
({
data
})
)
,
m_split_evenly
{
false
}
,
m_split_evenly
{
false
}
,
m_axis
{
axis
}
,
m_axis
{
axis
}
,
m_splits
{
splits
}
,
m_splits
{
splits
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/split.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
Split
:
public
ngraph
::
op
::
util
::
FusedOp
class
Split
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs a Split op that evenly divides the input tensor.
/// \brief Constructs a Split op that evenly divides the input tensor.
///
///
/// \param data - Node producing the input tensor
/// \param data - Node producing the input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/squared_difference.cpp
View file @
410d6e61
...
@@ -24,8 +24,10 @@
...
@@ -24,8 +24,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
SquaredDifference
::
type_name
{
"SquaredDifference"
};
op
::
SquaredDifference
::
SquaredDifference
(
const
shared_ptr
<
Node
>&
x1
,
const
shared_ptr
<
Node
>&
x2
)
op
::
SquaredDifference
::
SquaredDifference
(
const
shared_ptr
<
Node
>&
x1
,
const
shared_ptr
<
Node
>&
x2
)
:
FusedOp
(
"SquaredDifference"
,
{
x1
,
x2
}
)
:
FusedOp
(
check_single_output_args
({
x1
,
x2
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/squared_difference.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
SquaredDifference
:
public
ngraph
::
op
::
util
::
FusedOp
class
SquaredDifference
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
/// \brief Constructs the squared difference operation.
/// \brief Constructs the squared difference operation.
///
///
/// \param x1 First input tensor
/// \param x1 First input tensor
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/squeeze.cpp
View file @
410d6e61
...
@@ -26,8 +26,10 @@
...
@@ -26,8 +26,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
Squeeze
::
type_name
{
"Squeeze"
};
op
::
Squeeze
::
Squeeze
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
axes
)
op
::
Squeeze
::
Squeeze
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
axes
)
:
FusedOp
(
"Squeeze"
,
{
data
,
axes
}
)
:
FusedOp
(
check_single_output_args
({
data
,
axes
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/squeeze.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
Squeeze
:
public
ngraph
::
op
::
util
::
FusedOp
class
Squeeze
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
Squeeze
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
Squeeze
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
axes
);
const
std
::
shared_ptr
<
ngraph
::
Node
>&
axes
);
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/unsqueeze.cpp
View file @
410d6e61
...
@@ -26,8 +26,10 @@
...
@@ -26,8 +26,10 @@
using
namespace
std
;
using
namespace
std
;
using
namespace
ngraph
;
using
namespace
ngraph
;
const
string
op
::
Unsqueeze
::
type_name
{
"Unsqueeze"
};
op
::
Unsqueeze
::
Unsqueeze
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
axes
)
op
::
Unsqueeze
::
Unsqueeze
(
const
shared_ptr
<
Node
>&
data
,
const
shared_ptr
<
Node
>&
axes
)
:
FusedOp
(
"Unsqueeze"
,
{
data
,
axes
}
)
:
FusedOp
(
check_single_output_args
({
data
,
axes
})
)
{
{
constructor_validate_and_infer_types
();
constructor_validate_and_infer_types
();
}
}
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/op/fused/unsqueeze.hpp
View file @
410d6e61
...
@@ -30,6 +30,9 @@ namespace ngraph
...
@@ -30,6 +30,9 @@ namespace ngraph
class
Unsqueeze
:
public
ngraph
::
op
::
util
::
FusedOp
class
Unsqueeze
:
public
ngraph
::
op
::
util
::
FusedOp
{
{
public
:
public
:
NGRAPH_API
static
const
std
::
string
type_name
;
const
std
::
string
&
description
()
const
override
{
return
type_name
;
}
Unsqueeze
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
Unsqueeze
(
const
std
::
shared_ptr
<
ngraph
::
Node
>&
data
,
const
std
::
shared_ptr
<
ngraph
::
Node
>&
axes
);
const
std
::
shared_ptr
<
ngraph
::
Node
>&
axes
);
...
...
This diff is collapsed.
Click to expand it.
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