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
7a4cc10c
Unverified
Commit
7a4cc10c
authored
Aug 02, 2019
by
Scott Cyphers
Committed by
GitHub
Aug 02, 2019
Browse files
Options
Browse Files
Download
Plain Diff
Merge branch 'master' into pruthvi/cpu_static_backend
parents
3711118e
410d6e61
Hide 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 @
7a4cc10c
...
@@ -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
)
...
...
src/ngraph/op/fused/conv_fused.hpp
View file @
7a4cc10c
...
@@ -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
);
...
...
src/ngraph/op/fused/depth_to_space.cpp
View file @
7a4cc10c
...
@@ -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
();
...
...
src/ngraph/op/fused/depth_to_space.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/elu.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/elu.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/fake_quantize.cpp
View file @
7a4cc10c
...
@@ -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
();
...
...
src/ngraph/op/fused/fake_quantize.hpp
View file @
7a4cc10c
...
@@ -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.
///
///
...
...
src/ngraph/op/fused/gelu.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/gelu.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/gemm.cpp
View file @
7a4cc10c
...
@@ -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
}
...
...
src/ngraph/op/fused/gemm.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/grn.cpp
View file @
7a4cc10c
...
@@ -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
();
...
...
src/ngraph/op/fused/grn.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/group_conv_transpose.cpp
View file @
7a4cc10c
...
@@ -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
)
...
...
src/ngraph/op/fused/group_conv_transpose.hpp
View file @
7a4cc10c
...
@@ -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.
///
///
...
...
src/ngraph/op/fused/gru_cell.cpp
View file @
7a4cc10c
...
@@ -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
)}
...
...
src/ngraph/op/fused/gru_cell.hpp
View file @
7a4cc10c
...
@@ -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.
///
///
...
...
src/ngraph/op/fused/hard_sigmoid.cpp
View file @
7a4cc10c
...
@@ -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
)
{
{
...
...
src/ngraph/op/fused/hard_sigmoid.hpp
View file @
7a4cc10c
...
@@ -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.
...
...
src/ngraph/op/fused/leaky_relu.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/leaky_relu.hpp
View file @
7a4cc10c
...
@@ -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
);
...
...
src/ngraph/op/fused/lstm_cell.cpp
View file @
7a4cc10c
...
@@ -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
)}
...
...
src/ngraph/op/fused/lstm_cell.hpp
View file @
7a4cc10c
...
@@ -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.
///
///
...
...
src/ngraph/op/fused/mvn.cpp
View file @
7a4cc10c
...
@@ -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
}
...
...
src/ngraph/op/fused/mvn.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/normalize.cpp
View file @
7a4cc10c
...
@@ -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
}
...
...
src/ngraph/op/fused/normalize.hpp
View file @
7a4cc10c
...
@@ -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.
///
///
...
...
src/ngraph/op/fused/prelu.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/prelu.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/rnn_cell.cpp
View file @
7a4cc10c
...
@@ -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
)}
{
{
...
...
src/ngraph/op/fused/rnn_cell.hpp
View file @
7a4cc10c
...
@@ -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.
///
///
...
...
src/ngraph/op/fused/scale_shift.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/scale_shift.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/shuffle_channels.cpp
View file @
7a4cc10c
...
@@ -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
}
{
{
...
...
src/ngraph/op/fused/shuffle_channels.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/space_to_depth.cpp
View file @
7a4cc10c
...
@@ -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
();
...
...
src/ngraph/op/fused/space_to_depth.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/split.cpp
View file @
7a4cc10c
...
@@ -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
}
...
...
src/ngraph/op/fused/split.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/squared_difference.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/squared_difference.hpp
View file @
7a4cc10c
...
@@ -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
...
...
src/ngraph/op/fused/squeeze.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/squeeze.hpp
View file @
7a4cc10c
...
@@ -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
);
...
...
src/ngraph/op/fused/unsqueeze.cpp
View file @
7a4cc10c
...
@@ -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
();
}
}
...
...
src/ngraph/op/fused/unsqueeze.hpp
View file @
7a4cc10c
...
@@ -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
);
...
...
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