Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
2301_77200941
mindspore
提交
02a954ab
M
mindspore
项目概览
2301_77200941
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
02a954ab
编写于
8月 27, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
8月 27, 2020
浏览文件
操作
浏览文件
下载
差异文件
!5047 copy primitive
Merge pull request !5047 from lyvette/master
上级
e94416be
6f75b506
变更
120
隐藏空白更改
内联
并排
Showing
120 changed file
with
2956 addition
and
0 deletion
+2956
-0
mindspore/lite/src/ops/abs.h
mindspore/lite/src/ops/abs.h
+20
-0
mindspore/lite/src/ops/activation.h
mindspore/lite/src/ops/activation.h
+23
-0
mindspore/lite/src/ops/activation_grad.h
mindspore/lite/src/ops/activation_grad.h
+23
-0
mindspore/lite/src/ops/add.h
mindspore/lite/src/ops/add.h
+23
-0
mindspore/lite/src/ops/addn.h
mindspore/lite/src/ops/addn.h
+23
-0
mindspore/lite/src/ops/argmax.h
mindspore/lite/src/ops/argmax.h
+24
-0
mindspore/lite/src/ops/argmin.h
mindspore/lite/src/ops/argmin.h
+24
-0
mindspore/lite/src/ops/batch_norm.h
mindspore/lite/src/ops/batch_norm.h
+20
-0
mindspore/lite/src/ops/batch_to_space.h
mindspore/lite/src/ops/batch_to_space.h
+33
-0
mindspore/lite/src/ops/bias_add.h
mindspore/lite/src/ops/bias_add.h
+29
-0
mindspore/lite/src/ops/bias_grad.h
mindspore/lite/src/ops/bias_grad.h
+29
-0
mindspore/lite/src/ops/bn_grad_input.h
mindspore/lite/src/ops/bn_grad_input.h
+23
-0
mindspore/lite/src/ops/broadcast_to.h
mindspore/lite/src/ops/broadcast_to.h
+29
-0
mindspore/lite/src/ops/caffe_p_relu.h
mindspore/lite/src/ops/caffe_p_relu.h
+29
-0
mindspore/lite/src/ops/cast.h
mindspore/lite/src/ops/cast.h
+23
-0
mindspore/lite/src/ops/ceil.h
mindspore/lite/src/ops/ceil.h
+20
-0
mindspore/lite/src/ops/clip.h
mindspore/lite/src/ops/clip.h
+23
-0
mindspore/lite/src/ops/concat.h
mindspore/lite/src/ops/concat.h
+23
-0
mindspore/lite/src/ops/constant_of_shape.h
mindspore/lite/src/ops/constant_of_shape.h
+23
-0
mindspore/lite/src/ops/conv2d.h
mindspore/lite/src/ops/conv2d.h
+27
-0
mindspore/lite/src/ops/conv2d_grad_filter.h
mindspore/lite/src/ops/conv2d_grad_filter.h
+28
-0
mindspore/lite/src/ops/conv2d_grad_input.h
mindspore/lite/src/ops/conv2d_grad_input.h
+28
-0
mindspore/lite/src/ops/cos.h
mindspore/lite/src/ops/cos.h
+20
-0
mindspore/lite/src/ops/crop.h
mindspore/lite/src/ops/crop.h
+29
-0
mindspore/lite/src/ops/deconv2d.h
mindspore/lite/src/ops/deconv2d.h
+27
-0
mindspore/lite/src/ops/dedepthwise_conv2d.h
mindspore/lite/src/ops/dedepthwise_conv2d.h
+27
-0
mindspore/lite/src/ops/depth_to_space.h
mindspore/lite/src/ops/depth_to_space.h
+23
-0
mindspore/lite/src/ops/depthwise_conv2d.h
mindspore/lite/src/ops/depthwise_conv2d.h
+28
-0
mindspore/lite/src/ops/detection_post_process.h
mindspore/lite/src/ops/detection_post_process.h
+29
-0
mindspore/lite/src/ops/div.h
mindspore/lite/src/ops/div.h
+23
-0
mindspore/lite/src/ops/dropout.h
mindspore/lite/src/ops/dropout.h
+23
-0
mindspore/lite/src/ops/eltwise.h
mindspore/lite/src/ops/eltwise.h
+23
-0
mindspore/lite/src/ops/elu.h
mindspore/lite/src/ops/elu.h
+23
-0
mindspore/lite/src/ops/embedding_lookup.h
mindspore/lite/src/ops/embedding_lookup.h
+23
-0
mindspore/lite/src/ops/embedding_lookup_sparse.h
mindspore/lite/src/ops/embedding_lookup_sparse.h
+33
-0
mindspore/lite/src/ops/equal.h
mindspore/lite/src/ops/equal.h
+20
-0
mindspore/lite/src/ops/exp.h
mindspore/lite/src/ops/exp.h
+20
-0
mindspore/lite/src/ops/expand_dims.h
mindspore/lite/src/ops/expand_dims.h
+23
-0
mindspore/lite/src/ops/fake_quant_with_min_max_vars.h
mindspore/lite/src/ops/fake_quant_with_min_max_vars.h
+24
-0
mindspore/lite/src/ops/fill.h
mindspore/lite/src/ops/fill.h
+29
-0
mindspore/lite/src/ops/flatten.h
mindspore/lite/src/ops/flatten.h
+20
-0
mindspore/lite/src/ops/floor.h
mindspore/lite/src/ops/floor.h
+20
-0
mindspore/lite/src/ops/floor_div.h
mindspore/lite/src/ops/floor_div.h
+20
-0
mindspore/lite/src/ops/floor_mod.h
mindspore/lite/src/ops/floor_mod.h
+20
-0
mindspore/lite/src/ops/full_connection.h
mindspore/lite/src/ops/full_connection.h
+24
-0
mindspore/lite/src/ops/fused_batchnorm.h
mindspore/lite/src/ops/fused_batchnorm.h
+23
-0
mindspore/lite/src/ops/gather.h
mindspore/lite/src/ops/gather.h
+23
-0
mindspore/lite/src/ops/gather_nd.h
mindspore/lite/src/ops/gather_nd.h
+23
-0
mindspore/lite/src/ops/greater.h
mindspore/lite/src/ops/greater.h
+20
-0
mindspore/lite/src/ops/greater_equal.h
mindspore/lite/src/ops/greater_equal.h
+20
-0
mindspore/lite/src/ops/l2_norm.h
mindspore/lite/src/ops/l2_norm.h
+29
-0
mindspore/lite/src/ops/leaky_relu.h
mindspore/lite/src/ops/leaky_relu.h
+23
-0
mindspore/lite/src/ops/less.h
mindspore/lite/src/ops/less.h
+20
-0
mindspore/lite/src/ops/less_equal.h
mindspore/lite/src/ops/less_equal.h
+20
-0
mindspore/lite/src/ops/local_response_normalization.h
mindspore/lite/src/ops/local_response_normalization.h
+24
-0
mindspore/lite/src/ops/log.h
mindspore/lite/src/ops/log.h
+20
-0
mindspore/lite/src/ops/logical_and.h
mindspore/lite/src/ops/logical_and.h
+20
-0
mindspore/lite/src/ops/logical_not.h
mindspore/lite/src/ops/logical_not.h
+20
-0
mindspore/lite/src/ops/logical_or.h
mindspore/lite/src/ops/logical_or.h
+20
-0
mindspore/lite/src/ops/lrn.h
mindspore/lite/src/ops/lrn.h
+23
-0
mindspore/lite/src/ops/lstm.h
mindspore/lite/src/ops/lstm.h
+23
-0
mindspore/lite/src/ops/make_tuple.h
mindspore/lite/src/ops/make_tuple.h
+20
-0
mindspore/lite/src/ops/matmul.h
mindspore/lite/src/ops/matmul.h
+23
-0
mindspore/lite/src/ops/matrix_diag.h
mindspore/lite/src/ops/matrix_diag.h
+24
-0
mindspore/lite/src/ops/maximum.h
mindspore/lite/src/ops/maximum.h
+20
-0
mindspore/lite/src/ops/mean.h
mindspore/lite/src/ops/mean.h
+29
-0
mindspore/lite/src/ops/minimum.h
mindspore/lite/src/ops/minimum.h
+20
-0
mindspore/lite/src/ops/mul.h
mindspore/lite/src/ops/mul.h
+23
-0
mindspore/lite/src/ops/nchw2nhwc.h
mindspore/lite/src/ops/nchw2nhwc.h
+20
-0
mindspore/lite/src/ops/nhwc2nchw.h
mindspore/lite/src/ops/nhwc2nchw.h
+20
-0
mindspore/lite/src/ops/not_equal.h
mindspore/lite/src/ops/not_equal.h
+20
-0
mindspore/lite/src/ops/one_hot.h
mindspore/lite/src/ops/one_hot.h
+23
-0
mindspore/lite/src/ops/pad.h
mindspore/lite/src/ops/pad.h
+29
-0
mindspore/lite/src/ops/permute.h
mindspore/lite/src/ops/permute.h
+29
-0
mindspore/lite/src/ops/pooling.h
mindspore/lite/src/ops/pooling.h
+26
-0
mindspore/lite/src/ops/pooling_grad.h
mindspore/lite/src/ops/pooling_grad.h
+26
-0
mindspore/lite/src/ops/power.h
mindspore/lite/src/ops/power.h
+23
-0
mindspore/lite/src/ops/power_grad.h
mindspore/lite/src/ops/power_grad.h
+23
-0
mindspore/lite/src/ops/prelu.h
mindspore/lite/src/ops/prelu.h
+29
-0
mindspore/lite/src/ops/prior_box.h
mindspore/lite/src/ops/prior_box.h
+42
-0
mindspore/lite/src/ops/quant_dtype_cast.h
mindspore/lite/src/ops/quant_dtype_cast.h
+23
-0
mindspore/lite/src/ops/range.h
mindspore/lite/src/ops/range.h
+23
-0
mindspore/lite/src/ops/rank.h
mindspore/lite/src/ops/rank.h
+20
-0
mindspore/lite/src/ops/reduce.h
mindspore/lite/src/ops/reduce.h
+29
-0
mindspore/lite/src/ops/reshape.h
mindspore/lite/src/ops/reshape.h
+29
-0
mindspore/lite/src/ops/resize.h
mindspore/lite/src/ops/resize.h
+25
-0
mindspore/lite/src/ops/reverse.h
mindspore/lite/src/ops/reverse.h
+29
-0
mindspore/lite/src/ops/reverse_sequence.h
mindspore/lite/src/ops/reverse_sequence.h
+30
-0
mindspore/lite/src/ops/roi_pooling.h
mindspore/lite/src/ops/roi_pooling.h
+23
-0
mindspore/lite/src/ops/round.h
mindspore/lite/src/ops/round.h
+20
-0
mindspore/lite/src/ops/rsqrt.h
mindspore/lite/src/ops/rsqrt.h
+20
-0
mindspore/lite/src/ops/scale.h
mindspore/lite/src/ops/scale.h
+23
-0
mindspore/lite/src/ops/scatter_nd.h
mindspore/lite/src/ops/scatter_nd.h
+20
-0
mindspore/lite/src/ops/shape.h
mindspore/lite/src/ops/shape.h
+20
-0
mindspore/lite/src/ops/sin.h
mindspore/lite/src/ops/sin.h
+20
-0
mindspore/lite/src/ops/slice.h
mindspore/lite/src/ops/slice.h
+34
-0
mindspore/lite/src/ops/softmax.h
mindspore/lite/src/ops/softmax.h
+23
-0
mindspore/lite/src/ops/softmax_cross_entropy.h
mindspore/lite/src/ops/softmax_cross_entropy.h
+29
-0
mindspore/lite/src/ops/space_to_batch.h
mindspore/lite/src/ops/space_to_batch.h
+33
-0
mindspore/lite/src/ops/space_to_batch_nd.h
mindspore/lite/src/ops/space_to_batch_nd.h
+33
-0
mindspore/lite/src/ops/space_to_depth.h
mindspore/lite/src/ops/space_to_depth.h
+23
-0
mindspore/lite/src/ops/sparse_to_dense.h
mindspore/lite/src/ops/sparse_to_dense.h
+38
-0
mindspore/lite/src/ops/split.h
mindspore/lite/src/ops/split.h
+29
-0
mindspore/lite/src/ops/sqrt.h
mindspore/lite/src/ops/sqrt.h
+20
-0
mindspore/lite/src/ops/square.h
mindspore/lite/src/ops/square.h
+20
-0
mindspore/lite/src/ops/squared_difference.h
mindspore/lite/src/ops/squared_difference.h
+20
-0
mindspore/lite/src/ops/squeeze.h
mindspore/lite/src/ops/squeeze.h
+29
-0
mindspore/lite/src/ops/stack.h
mindspore/lite/src/ops/stack.h
+29
-0
mindspore/lite/src/ops/strided_slice.h
mindspore/lite/src/ops/strided_slice.h
+43
-0
mindspore/lite/src/ops/sub.h
mindspore/lite/src/ops/sub.h
+23
-0
mindspore/lite/src/ops/tile.h
mindspore/lite/src/ops/tile.h
+33
-0
mindspore/lite/src/ops/topk.h
mindspore/lite/src/ops/topk.h
+23
-0
mindspore/lite/src/ops/transpose.h
mindspore/lite/src/ops/transpose.h
+29
-0
mindspore/lite/src/ops/tuple_get_item.h
mindspore/lite/src/ops/tuple_get_item.h
+20
-0
mindspore/lite/src/ops/unique.h
mindspore/lite/src/ops/unique.h
+23
-0
mindspore/lite/src/ops/unsqueeze.h
mindspore/lite/src/ops/unsqueeze.h
+29
-0
mindspore/lite/src/ops/unstack.h
mindspore/lite/src/ops/unstack.h
+23
-0
mindspore/lite/src/ops/upsample.h
mindspore/lite/src/ops/upsample.h
+29
-0
mindspore/lite/src/ops/where.h
mindspore/lite/src/ops/where.h
+29
-0
mindspore/lite/src/ops/zeros_like.h
mindspore/lite/src/ops/zeros_like.h
+20
-0
未找到文件。
mindspore/lite/src/ops/abs.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Abs : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Abs : public ArithmeticSelf {
explicit
Abs
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Abs
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Abs
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Abs
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateAbs
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Abs
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/activation.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Activation : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Activation : public PrimitiveC {
void
SetAlpha
(
float
alpha
);
void
SetAlpha
(
float
alpha
);
#else
#else
explicit
Activation
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Activation
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Activation
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateActivation
(
fbb
,
attr
->
type
(),
attr
->
alpha
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Activation
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetType
()
const
;
int
GetType
()
const
;
float
GetAlpha
()
const
;
float
GetAlpha
()
const
;
...
...
mindspore/lite/src/ops/activation_grad.h
浏览文件 @
02a954ab
...
@@ -34,6 +34,29 @@ class ActivationGrad : public PrimitiveC {
...
@@ -34,6 +34,29 @@ class ActivationGrad : public PrimitiveC {
void
SetType
(
int
type
);
void
SetType
(
int
type
);
#else
#else
explicit
ActivationGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ActivationGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_ActivationGrad
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateActivationGrad
(
fbb
,
attr
->
type
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ActivationGrad
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetType
()
const
;
int
GetType
()
const
;
};
};
...
...
mindspore/lite/src/ops/add.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Add : public Arithmetic {
...
@@ -35,6 +35,29 @@ class Add : public Arithmetic {
void
SetActivationType
(
int
activation_type
);
void
SetActivationType
(
int
activation_type
);
#else
#else
explicit
Add
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Add
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Add
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateAdd
(
fbb
,
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Add
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetActivationType
()
const
;
int
GetActivationType
()
const
;
};
};
...
...
mindspore/lite/src/ops/addn.h
浏览文件 @
02a954ab
...
@@ -34,6 +34,29 @@ class AddN : public PrimitiveC {
...
@@ -34,6 +34,29 @@ class AddN : public PrimitiveC {
void
SetN
(
int
n
);
void
SetN
(
int
n
);
#else
#else
explicit
AddN
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
AddN
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_AddN
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateAddN
(
fbb
,
attr
->
N
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_AddN
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetN
()
const
;
int
GetN
()
const
;
...
...
mindspore/lite/src/ops/argmax.h
浏览文件 @
02a954ab
...
@@ -38,6 +38,30 @@ class ArgMax : public PrimitiveC {
...
@@ -38,6 +38,30 @@ class ArgMax : public PrimitiveC {
void
SetAxisType
(
int
axis_type
);
void
SetAxisType
(
int
axis_type
);
#else
#else
explicit
ArgMax
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ArgMax
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_ArgMax
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateArgMax
(
fbb
,
attr
->
axis
(),
attr
->
outMaxValue
(),
attr
->
topK
(),
attr
->
keepDims
(),
attr
->
axisType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ArgMax
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetAxis
()
const
;
int
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/argmin.h
浏览文件 @
02a954ab
...
@@ -38,6 +38,30 @@ class ArgMin : public PrimitiveC {
...
@@ -38,6 +38,30 @@ class ArgMin : public PrimitiveC {
void
SetAxisType
(
int
axis_type
);
void
SetAxisType
(
int
axis_type
);
#else
#else
explicit
ArgMin
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ArgMin
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_ArgMin
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateArgMin
(
fbb
,
attr
->
axis
(),
attr
->
outMaxValue
(),
attr
->
topK
(),
attr
->
keepDims
(),
attr
->
axisType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ArgMin
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetAxis
()
const
;
int
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/batch_norm.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,26 @@ class BatchNorm : public PrimitiveC {
...
@@ -35,6 +35,26 @@ class BatchNorm : public PrimitiveC {
void
SetEpsilon
(
float
epsilon
);
void
SetEpsilon
(
float
epsilon
);
#else
#else
explicit
BatchNorm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
BatchNorm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateBatchNorm
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_BatchNorm
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetEpsilon
()
const
;
float
GetEpsilon
()
const
;
};
};
...
...
mindspore/lite/src/ops/batch_to_space.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,38 @@ class BatchToSpace : public PrimitiveC {
...
@@ -35,6 +36,38 @@ class BatchToSpace : public PrimitiveC {
void
SetCrops
(
const
std
::
vector
<
int
>
&
crops
);
void
SetCrops
(
const
std
::
vector
<
int
>
&
crops
);
#else
#else
explicit
BatchToSpace
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
BatchToSpace
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_BatchToSpace
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
blockShape
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
blockShape
()
->
size
());
i
++
)
{
blockShape
->
push_back
(
attr
->
blockShape
()
->
data
()[
i
]);
}
auto
crops
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
crops
()
->
size
());
i
++
)
{
crops
->
push_back
(
attr
->
crops
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateBatchToSpaceDirect
(
fbb
,
blockShape
.
release
(),
crops
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_BatchToSpace
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetBlockShape
()
const
;
std
::
vector
<
int
>
GetBlockShape
()
const
;
...
...
mindspore/lite/src/ops/bias_add.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class BiasAdd : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class BiasAdd : public PrimitiveC {
void
SetAxis
(
const
std
::
vector
<
int
>
&
axis
);
void
SetAxis
(
const
std
::
vector
<
int
>
&
axis
);
#else
#else
explicit
BiasAdd
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
BiasAdd
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_BiasAdd
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateBiasAddDirect
(
fbb
,
axis
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_BiasAdd
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
};
};
...
...
mindspore/lite/src/ops/bias_grad.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class BiasGrad : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class BiasGrad : public PrimitiveC {
#else
#else
explicit
BiasGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
BiasGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_BiasGrad
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateBiasGradDirect
(
fbb
,
axis
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_BiasGrad
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
};
};
...
...
mindspore/lite/src/ops/bn_grad_input.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class BNGradInput : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class BNGradInput : public PrimitiveC {
void
SetChannels
(
int
channels
);
void
SetChannels
(
int
channels
);
#else
#else
explicit
BNGradInput
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
BNGradInput
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_BNGradInput
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateBNGradInput
(
fbb
,
attr
->
eps
(),
attr
->
channels
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_BNGradInput
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetEps
()
const
;
float
GetEps
()
const
;
int
GetChannels
()
const
;
int
GetChannels
()
const
;
...
...
mindspore/lite/src/ops/broadcast_to.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class BroadcastTo : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class BroadcastTo : public PrimitiveC {
#else
#else
explicit
BroadcastTo
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
BroadcastTo
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_BroadcastTo
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
dst_shape
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
dst_shape
()
->
size
());
i
++
)
{
dst_shape
->
push_back
(
attr
->
dst_shape
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateBroadcastToDirect
(
fbb
,
dst_shape
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_BroadcastTo
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetDstShape
()
const
;
std
::
vector
<
int
>
GetDstShape
()
const
;
...
...
mindspore/lite/src/ops/caffe_p_relu.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/activation.h"
#include "src/ops/activation.h"
...
@@ -35,6 +36,34 @@ class CaffePReLU : public Activation {
...
@@ -35,6 +36,34 @@ class CaffePReLU : public Activation {
#else
#else
explicit
CaffePReLU
(
schema
::
Primitive
*
primitive
)
:
Activation
(
primitive
)
{}
explicit
CaffePReLU
(
schema
::
Primitive
*
primitive
)
:
Activation
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_CaffePReLU
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
slope
=
std
::
make_unique
<
std
::
vector
<
float
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
slope
()
->
size
());
i
++
)
{
slope
->
push_back
(
attr
->
slope
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateCaffePReLUDirect
(
fbb
,
attr
->
channelShared
(),
slope
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_CaffePReLU
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
bool
GetChannelShared
()
const
;
bool
GetChannelShared
()
const
;
};
};
...
...
mindspore/lite/src/ops/cast.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Cast : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Cast : public PrimitiveC {
void
SetDstT
(
int
dst_t
);
void
SetDstT
(
int
dst_t
);
#else
#else
explicit
Cast
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Cast
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Cast
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateCast
(
fbb
,
attr
->
srcT
(),
attr
->
dstT
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Cast
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetSrcT
()
const
;
int
GetSrcT
()
const
;
...
...
mindspore/lite/src/ops/ceil.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Ceil : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Ceil : public ArithmeticSelf {
explicit
Ceil
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Ceil
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Ceil
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Ceil
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateCeil
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Ceil
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/clip.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Clip : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Clip : public PrimitiveC {
void
SetMin
(
float
min
);
void
SetMin
(
float
min
);
#else
#else
explicit
Clip
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Clip
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Clip
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateClip
(
fbb
,
attr
->
max
(),
attr
->
min
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Clip
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetMax
()
const
;
float
GetMax
()
const
;
float
GetMin
()
const
;
float
GetMin
()
const
;
...
...
mindspore/lite/src/ops/concat.h
浏览文件 @
02a954ab
...
@@ -36,6 +36,29 @@ class Concat : public PrimitiveC {
...
@@ -36,6 +36,29 @@ class Concat : public PrimitiveC {
void
SetN
(
int
n
);
void
SetN
(
int
n
);
#else
#else
explicit
Concat
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Concat
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Concat
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateConcat
(
fbb
,
attr
->
axis
(),
attr
->
n
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Concat
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetAxis
()
const
;
int
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/constant_of_shape.h
浏览文件 @
02a954ab
...
@@ -34,6 +34,29 @@ class ConstantOfShape : public PrimitiveC {
...
@@ -34,6 +34,29 @@ class ConstantOfShape : public PrimitiveC {
void
SetValue
(
float
value
);
void
SetValue
(
float
value
);
#else
#else
explicit
ConstantOfShape
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ConstantOfShape
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_ConstantOfShape
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateConstantOfShape
(
fbb
,
attr
->
value
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ConstantOfShape
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
float
GetValue
()
const
;
float
GetValue
()
const
;
...
...
mindspore/lite/src/ops/conv2d.h
浏览文件 @
02a954ab
...
@@ -64,6 +64,33 @@ class Conv2D : public PrimitiveC {
...
@@ -64,6 +64,33 @@ class Conv2D : public PrimitiveC {
public:
public:
explicit
Conv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Conv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Conv2D
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateConv2D
(
fbb
,
attr
->
format
(),
attr
->
group
(),
attr
->
channelIn
(),
attr
->
channelOut
(),
attr
->
kernelW
(),
attr
->
kernelH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
dilateW
(),
attr
->
dilateH
(),
attr
->
hasBias
(),
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Conv2D
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
public:
public:
...
...
mindspore/lite/src/ops/conv2d_grad_filter.h
浏览文件 @
02a954ab
...
@@ -50,6 +50,34 @@ class Conv2DGradFilter : public PrimitiveC {
...
@@ -50,6 +50,34 @@ class Conv2DGradFilter : public PrimitiveC {
void
SetActivationType
(
int
activation_type
);
void
SetActivationType
(
int
activation_type
);
#else
#else
explicit
Conv2DGradFilter
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Conv2DGradFilter
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Conv2DGradFilter
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateConv2DGradFilter
(
fbb
,
attr
->
format
(),
attr
->
group
(),
attr
->
channelIn
(),
attr
->
channelOut
(),
attr
->
kernelW
(),
attr
->
kernelH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
dilateW
(),
attr
->
dilateH
(),
attr
->
hasBias
(),
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Conv2DGradFilter
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetFormat
()
const
;
int
GetFormat
()
const
;
int
GetGroup
()
const
;
int
GetGroup
()
const
;
...
...
mindspore/lite/src/ops/conv2d_grad_input.h
浏览文件 @
02a954ab
...
@@ -50,6 +50,34 @@ class Conv2DGradInput : public PrimitiveC {
...
@@ -50,6 +50,34 @@ class Conv2DGradInput : public PrimitiveC {
void
SetActivationType
(
int
activation_type
);
void
SetActivationType
(
int
activation_type
);
#else
#else
explicit
Conv2DGradInput
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Conv2DGradInput
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Conv2DGradInput
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateConv2DGradInput
(
fbb
,
attr
->
format
(),
attr
->
group
(),
attr
->
channelIn
(),
attr
->
channelOut
(),
attr
->
kernelW
(),
attr
->
kernelH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
dilateW
(),
attr
->
dilateH
(),
attr
->
hasBias
(),
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Conv2DGradInput
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetFormat
()
const
;
int
GetFormat
()
const
;
int
GetGroup
()
const
;
int
GetGroup
()
const
;
...
...
mindspore/lite/src/ops/cos.h
浏览文件 @
02a954ab
...
@@ -32,6 +32,26 @@ class Cos : public ArithmeticSelf {
...
@@ -32,6 +32,26 @@ class Cos : public ArithmeticSelf {
explicit
Cos
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Cos
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Cos
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Cos
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateCos
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Cos
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/crop.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class Crop : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class Crop : public PrimitiveC {
void
SetOffsets
(
const
std
::
vector
<
int64_t
>
&
offsets
);
void
SetOffsets
(
const
std
::
vector
<
int64_t
>
&
offsets
);
#else
#else
explicit
Crop
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Crop
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Crop
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
offsets
=
std
::
make_unique
<
std
::
vector
<
int64_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
offsets
()
->
size
());
i
++
)
{
offsets
->
push_back
(
attr
->
offsets
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateCropDirect
(
fbb
,
attr
->
axis
(),
offsets
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Crop
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int64_t
GetAxis
()
const
;
int64_t
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/deconv2d.h
浏览文件 @
02a954ab
...
@@ -50,6 +50,33 @@ class DeConv2D : public PrimitiveC {
...
@@ -50,6 +50,33 @@ class DeConv2D : public PrimitiveC {
void
SetActivationType
(
int
activation_type
);
void
SetActivationType
(
int
activation_type
);
#else
#else
explicit
DeConv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
DeConv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_DeConv2D
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateDeConv2D
(
fbb
,
attr
->
format
(),
attr
->
group
(),
attr
->
channelIn
(),
attr
->
channelOut
(),
attr
->
kernelW
(),
attr
->
kernelH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
dilateW
(),
attr
->
dilateH
(),
attr
->
hasBias
(),
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_DeConv2D
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetFormat
()
const
;
int
GetFormat
()
const
;
...
...
mindspore/lite/src/ops/dedepthwise_conv2d.h
浏览文件 @
02a954ab
...
@@ -49,6 +49,33 @@ class DeDepthwiseConv2D : public PrimitiveC {
...
@@ -49,6 +49,33 @@ class DeDepthwiseConv2D : public PrimitiveC {
void
SetActivationType
(
int
activation_type
);
void
SetActivationType
(
int
activation_type
);
#else
#else
explicit
DeDepthwiseConv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
DeDepthwiseConv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_DeDepthwiseConv2D
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateDeDepthwiseConv2D
(
fbb
,
attr
->
format
(),
attr
->
channelIn
(),
attr
->
channelMultiplier
(),
attr
->
kernelW
(),
attr
->
kernelH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
dilateW
(),
attr
->
dilateH
(),
attr
->
hasBias
(),
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_DeDepthwiseConv2D
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetFormat
()
const
;
int
GetFormat
()
const
;
...
...
mindspore/lite/src/ops/depth_to_space.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class DepthToSpace : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class DepthToSpace : public PrimitiveC {
void
SetFormat
(
int
format
);
void
SetFormat
(
int
format
);
#else
#else
explicit
DepthToSpace
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
DepthToSpace
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_DepthToSpace
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateDepthToSpace
(
fbb
,
attr
->
blockSize
(),
attr
->
format
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_DepthToSpace
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetBlockSize
()
const
;
int
GetBlockSize
()
const
;
...
...
mindspore/lite/src/ops/depthwise_conv2d.h
浏览文件 @
02a954ab
...
@@ -59,6 +59,34 @@ class DepthwiseConv2D : public PrimitiveC {
...
@@ -59,6 +59,34 @@ class DepthwiseConv2D : public PrimitiveC {
public:
public:
explicit
DepthwiseConv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
DepthwiseConv2D
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_DepthwiseConv2D
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateDepthwiseConv2D
(
fbb
,
attr
->
format
(),
attr
->
channelIn
(),
attr
->
channelMultiplier
(),
attr
->
kernelW
(),
attr
->
kernelH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
dilateW
(),
attr
->
dilateH
(),
attr
->
hasBias
(),
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_DepthwiseConv2D
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
public:
public:
...
...
mindspore/lite/src/ops/detection_post_process.h
浏览文件 @
02a954ab
...
@@ -46,6 +46,35 @@ class DetectionPostProcess : public PrimitiveC {
...
@@ -46,6 +46,35 @@ class DetectionPostProcess : public PrimitiveC {
void
SetUseRegularNms
(
bool
use_regular_nms
);
void
SetUseRegularNms
(
bool
use_regular_nms
);
#else
#else
explicit
DetectionPostProcess
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
DetectionPostProcess
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_DetectionPostProcess
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateDetectionPostProcess
(
fbb
,
attr
->
format
(),
attr
->
inputSize
(),
attr
->
hScale
(),
attr
->
wScale
(),
attr
->
xScale
(),
attr
->
yScale
(),
attr
->
NmsIouThreshold
(),
attr
->
NmsScoreThreshold
(),
attr
->
MaxDetections
(),
attr
->
DetectionsPreClass
(),
attr
->
MaxClassesPreDetection
(),
attr
->
NumClasses
(),
attr
->
UseRegularNms
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_DetectionPostProcess
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetFormat
()
const
;
int
GetFormat
()
const
;
int
GetInputSize
()
const
;
int
GetInputSize
()
const
;
...
...
mindspore/lite/src/ops/div.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Div : public Arithmetic {
...
@@ -35,6 +35,29 @@ class Div : public Arithmetic {
#else
#else
explicit
Div
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Div
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Div
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateDiv
(
fbb
,
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Div
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetActivationType
()
const
;
int
GetActivationType
()
const
;
};
};
...
...
mindspore/lite/src/ops/dropout.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Dropout : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Dropout : public PrimitiveC {
#else
#else
explicit
Dropout
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Dropout
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Dropout
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateDropout
(
fbb
,
attr
->
ratio
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Dropout
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetRatio
()
const
;
float
GetRatio
()
const
;
};
};
...
...
mindspore/lite/src/ops/eltwise.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Eltwise : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Eltwise : public PrimitiveC {
#else
#else
explicit
Eltwise
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Eltwise
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Eltwise
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateEltwise
(
fbb
,
attr
->
mode
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Eltwise
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetMode
()
const
;
int
GetMode
()
const
;
};
};
...
...
mindspore/lite/src/ops/elu.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Elu : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Elu : public PrimitiveC {
#else
#else
explicit
Elu
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Elu
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Elu
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateElu
(
fbb
,
attr
->
alpha
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Elu
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetAlpha
()
const
;
float
GetAlpha
()
const
;
};
};
...
...
mindspore/lite/src/ops/embedding_lookup.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class EmbeddingLookup : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class EmbeddingLookup : public PrimitiveC {
#else
#else
explicit
EmbeddingLookup
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
EmbeddingLookup
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_EmbeddingLookup
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateEmbeddingLookup
(
fbb
,
attr
->
maxNorm
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_EmbeddingLookup
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
float
GetMaxNorm
()
const
;
float
GetMaxNorm
()
const
;
...
...
mindspore/lite/src/ops/embedding_lookup_sparse.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,38 @@ class EmbeddingLookupSparse : public PrimitiveC {
...
@@ -36,6 +37,38 @@ class EmbeddingLookupSparse : public PrimitiveC {
void
SetMaxNortm
(
float
max_nortm
);
void
SetMaxNortm
(
float
max_nortm
);
#else
#else
explicit
EmbeddingLookupSparse
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
EmbeddingLookupSparse
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_EmbeddingLookupSparse
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
spIds
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
spIds
()
->
size
());
i
++
)
{
spIds
->
push_back
(
attr
->
spIds
()
->
data
()[
i
]);
}
auto
spWeights
=
std
::
make_unique
<
std
::
vector
<
float
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
spWeights
()
->
size
());
i
++
)
{
spWeights
->
push_back
(
attr
->
spWeights
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateEmbeddingLookupSparseDirect
(
fbb
,
spIds
.
release
(),
spWeights
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_EmbeddingLookupSparse
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetSpIds
()
const
;
std
::
vector
<
int
>
GetSpIds
()
const
;
std
::
vector
<
float
>
GetSpWeights
()
const
;
std
::
vector
<
float
>
GetSpWeights
()
const
;
...
...
mindspore/lite/src/ops/equal.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Equal : public Arithmetic {
...
@@ -33,6 +33,26 @@ class Equal : public Arithmetic {
explicit
Equal
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Equal
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
Equal
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Equal
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateEqual
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Equal
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/exp.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Exp : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Exp : public ArithmeticSelf {
explicit
Exp
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Exp
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Exp
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Exp
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateExp
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Exp
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/expand_dims.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class ExpandDims : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class ExpandDims : public PrimitiveC {
#else
#else
explicit
ExpandDims
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ExpandDims
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_ExpandDims
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateExpandDims
(
fbb
,
attr
->
dim
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ExpandDims
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetDim
()
const
;
int
GetDim
()
const
;
...
...
mindspore/lite/src/ops/fake_quant_with_min_max_vars.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,30 @@ class FakeQuantWithMinMaxVars : public PrimitiveC {
...
@@ -35,6 +35,30 @@ class FakeQuantWithMinMaxVars : public PrimitiveC {
void
SetNumBits
(
int
num_bits
);
void
SetNumBits
(
int
num_bits
);
#else
#else
explicit
FakeQuantWithMinMaxVars
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
FakeQuantWithMinMaxVars
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_FakeQuantWithMinMaxVars
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateFakeQuantWithMinMaxVars
(
fbb
,
attr
->
narrowRange
(),
attr
->
numBits
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_FakeQuantWithMinMaxVars
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
bool
GetNarrowRange
()
const
;
bool
GetNarrowRange
()
const
;
int
GetNumBits
()
const
;
int
GetNumBits
()
const
;
...
...
mindspore/lite/src/ops/fill.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class Fill : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class Fill : public PrimitiveC {
#else
#else
explicit
Fill
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Fill
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Fill
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
dims
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
dims
()
->
size
());
i
++
)
{
dims
->
push_back
(
attr
->
dims
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateFillDirect
(
fbb
,
dims
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Fill
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetDims
()
const
;
std
::
vector
<
int
>
GetDims
()
const
;
...
...
mindspore/lite/src/ops/flatten.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Flatten : public PrimitiveC {
...
@@ -33,6 +33,26 @@ class Flatten : public PrimitiveC {
explicit
Flatten
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Flatten
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
Flatten
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Flatten
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateFlatten
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Flatten
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
...
...
mindspore/lite/src/ops/floor.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Floor : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Floor : public ArithmeticSelf {
explicit
Floor
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Floor
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Floor
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Floor
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateFloor
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Floor
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/floor_div.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class FloorDiv : public Arithmetic {
...
@@ -33,6 +33,26 @@ class FloorDiv : public Arithmetic {
explicit
FloorDiv
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
FloorDiv
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
FloorDiv
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
FloorDiv
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateFloorDiv
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_FloorDiv
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/floor_mod.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class FloorMod : public Arithmetic {
...
@@ -33,6 +33,26 @@ class FloorMod : public Arithmetic {
explicit
FloorMod
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
FloorMod
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
FloorMod
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
FloorMod
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateFloorMod
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_FloorMod
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/full_connection.h
浏览文件 @
02a954ab
...
@@ -37,6 +37,30 @@ class FullConnection : public PrimitiveC {
...
@@ -37,6 +37,30 @@ class FullConnection : public PrimitiveC {
void
SetActivationType
(
int
activationType
);
void
SetActivationType
(
int
activationType
);
#else
#else
explicit
FullConnection
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
FullConnection
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_FullConnection
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateFullConnection
(
fbb
,
attr
->
hasBias
(),
attr
->
axis
(),
attr
->
useAxis
(),
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_FullConnection
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
bool
GetHasBias
()
const
;
bool
GetHasBias
()
const
;
...
...
mindspore/lite/src/ops/fused_batchnorm.h
浏览文件 @
02a954ab
...
@@ -36,6 +36,29 @@ class FusedBatchNorm : public PrimitiveC {
...
@@ -36,6 +36,29 @@ class FusedBatchNorm : public PrimitiveC {
void
SetSpatial
(
int
spatial
);
void
SetSpatial
(
int
spatial
);
#else
#else
explicit
FusedBatchNorm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
FusedBatchNorm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_FusedBatchNorm
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateFusedBatchNorm
(
fbb
,
attr
->
epsilon
(),
attr
->
momentum
(),
attr
->
spatial
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_FusedBatchNorm
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetEpsilon
()
const
;
float
GetEpsilon
()
const
;
float
GetMomentum
()
const
;
float
GetMomentum
()
const
;
...
...
mindspore/lite/src/ops/gather.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Gather : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Gather : public PrimitiveC {
void
SetBatchDims
(
int
batch_dims
);
void
SetBatchDims
(
int
batch_dims
);
#else
#else
explicit
Gather
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Gather
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Gather
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateGather
(
fbb
,
attr
->
axis
(),
attr
->
batchDims
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Gather
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetAxis
()
const
;
int
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/gather_nd.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class GatherNd : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class GatherNd : public PrimitiveC {
#else
#else
explicit
GatherNd
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
GatherNd
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_GatherNd
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateGatherNd
(
fbb
,
attr
->
batchDims
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_GatherNd
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetBatchDims
()
const
;
int
GetBatchDims
()
const
;
...
...
mindspore/lite/src/ops/greater.h
浏览文件 @
02a954ab
...
@@ -32,6 +32,26 @@ class Greater : public Arithmetic {
...
@@ -32,6 +32,26 @@ class Greater : public Arithmetic {
explicit
Greater
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Greater
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
Greater
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Greater
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateGreater
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Greater
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/greater_equal.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class GreaterEqual : public Arithmetic {
...
@@ -33,6 +33,26 @@ class GreaterEqual : public Arithmetic {
explicit
GreaterEqual
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
GreaterEqual
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
GreaterEqual
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
GreaterEqual
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateGreaterEqual
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_GreaterEqual
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/l2_norm.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class L2Norm : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class L2Norm : public PrimitiveC {
void
SetEpsilon
(
float
epsilon
);
void
SetEpsilon
(
float
epsilon
);
#else
#else
explicit
L2Norm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
L2Norm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_L2Norm
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateL2NormDirect
(
fbb
,
axis
.
release
(),
attr
->
epsilon
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_L2Norm
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
float
GetEpsilon
()
const
;
float
GetEpsilon
()
const
;
...
...
mindspore/lite/src/ops/leaky_relu.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class LeakyReLU : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class LeakyReLU : public PrimitiveC {
#else
#else
explicit
LeakyReLU
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
LeakyReLU
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_LeakyReLU
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateLeakyReLU
(
fbb
,
attr
->
negativeSlope
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_LeakyReLU
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetNegativeSlope
()
const
;
float
GetNegativeSlope
()
const
;
};
};
...
...
mindspore/lite/src/ops/less.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Less : public Arithmetic {
...
@@ -33,6 +33,26 @@ class Less : public Arithmetic {
explicit
Less
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Less
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
Less
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Less
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateLess
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Less
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/less_equal.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class LessEqual : public Arithmetic {
...
@@ -33,6 +33,26 @@ class LessEqual : public Arithmetic {
explicit
LessEqual
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
LessEqual
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
LessEqual
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
LessEqual
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateLessEqual
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_LessEqual
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/local_response_normalization.h
浏览文件 @
02a954ab
...
@@ -37,6 +37,30 @@ class LocalResponseNormalization : public PrimitiveC {
...
@@ -37,6 +37,30 @@ class LocalResponseNormalization : public PrimitiveC {
void
SetBeta
(
float
beta
);
void
SetBeta
(
float
beta
);
#else
#else
explicit
LocalResponseNormalization
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
LocalResponseNormalization
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_LocalResponseNormalization
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateLocalResponseNormalization
(
fbb
,
attr
->
depth_radius
(),
attr
->
bias
(),
attr
->
alpha
(),
attr
->
beta
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_LocalResponseNormalization
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetDepthRadius
()
const
;
int
GetDepthRadius
()
const
;
float
GetBias
()
const
;
float
GetBias
()
const
;
...
...
mindspore/lite/src/ops/log.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Log : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Log : public ArithmeticSelf {
explicit
Log
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Log
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Log
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Log
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateLog
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Log
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/logical_and.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class LogicalAnd : public Arithmetic {
...
@@ -33,6 +33,26 @@ class LogicalAnd : public Arithmetic {
explicit
LogicalAnd
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
LogicalAnd
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
LogicalAnd
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
LogicalAnd
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateLogicalAnd
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_LogicalAnd
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/logical_not.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class LogicalNot : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class LogicalNot : public ArithmeticSelf {
explicit
LogicalNot
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
LogicalNot
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
LogicalNot
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
LogicalNot
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateLogicalNot
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_LogicalNot
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/logical_or.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class LogicalOr : public Arithmetic {
...
@@ -33,6 +33,26 @@ class LogicalOr : public Arithmetic {
explicit
LogicalOr
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
LogicalOr
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
LogicalOr
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
LogicalOr
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateLogicalOr
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_LogicalOr
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/lrn.h
浏览文件 @
02a954ab
...
@@ -37,6 +37,29 @@ class Lrn : public PrimitiveC {
...
@@ -37,6 +37,29 @@ class Lrn : public PrimitiveC {
void
SetSize
(
int
size
);
void
SetSize
(
int
size
);
#else
#else
explicit
Lrn
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Lrn
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Lrn
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateLrn
(
fbb
,
attr
->
alpha
(),
attr
->
beta
(),
attr
->
bias
(),
attr
->
size
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Lrn
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetAlpha
()
const
;
float
GetAlpha
()
const
;
float
GetBeta
()
const
;
float
GetBeta
()
const
;
...
...
mindspore/lite/src/ops/lstm.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Lstm : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Lstm : public PrimitiveC {
#else
#else
explicit
Lstm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Lstm
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Lstm
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateLstm
(
fbb
,
attr
->
bidirection
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Lstm
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
bool
GetBidirection
()
const
;
bool
GetBidirection
()
const
;
...
...
mindspore/lite/src/ops/make_tuple.h
浏览文件 @
02a954ab
...
@@ -30,6 +30,26 @@ class MakeTuple : public PrimitiveC {
...
@@ -30,6 +30,26 @@ class MakeTuple : public PrimitiveC {
int
UnPackAttr
(
const
Primitive
&
prim
,
const
std
::
vector
<
AnfNodePtr
>
&
inputs
);
int
UnPackAttr
(
const
Primitive
&
prim
,
const
std
::
vector
<
AnfNodePtr
>
&
inputs
);
#else
#else
explicit
MakeTuple
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
MakeTuple
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateMakeTuple
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_MakeTuple
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/matmul.h
浏览文件 @
02a954ab
...
@@ -44,6 +44,29 @@ class MatMul : public PrimitiveC {
...
@@ -44,6 +44,29 @@ class MatMul : public PrimitiveC {
public:
public:
explicit
MatMul
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
MatMul
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_MatMul
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateMatMul
(
fbb
,
attr
->
transposeA
(),
attr
->
transposeB
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_MatMul
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
public:
public:
...
...
mindspore/lite/src/ops/matrix_diag.h
浏览文件 @
02a954ab
...
@@ -37,6 +37,30 @@ class MatrixDiag : public PrimitiveC {
...
@@ -37,6 +37,30 @@ class MatrixDiag : public PrimitiveC {
void
SetPaddingValue
(
float
padding_value
);
void
SetPaddingValue
(
float
padding_value
);
#else
#else
explicit
MatrixDiag
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
MatrixDiag
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_MatrixDiag
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateMatrixDiag
(
fbb
,
attr
->
k
(),
attr
->
numRows
(),
attr
->
numCols
(),
attr
->
paddingValue
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_MatrixDiag
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetK
()
const
;
int
GetK
()
const
;
int
GetNumRows
()
const
;
int
GetNumRows
()
const
;
...
...
mindspore/lite/src/ops/maximum.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Maximum : public Arithmetic {
...
@@ -33,6 +33,26 @@ class Maximum : public Arithmetic {
explicit
Maximum
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Maximum
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
Maximum
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Maximum
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateMaximum
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Maximum
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/mean.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class Mean : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class Mean : public PrimitiveC {
void
SetKeepDims
(
bool
keep_dims
);
void
SetKeepDims
(
bool
keep_dims
);
#else
#else
explicit
Mean
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Mean
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Mean
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateMeanDirect
(
fbb
,
axis
.
release
(),
attr
->
keepDims
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Mean
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/minimum.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Minimum : public Arithmetic {
...
@@ -33,6 +33,26 @@ class Minimum : public Arithmetic {
explicit
Minimum
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Minimum
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
Minimum
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Minimum
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateMinimum
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Minimum
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/mul.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Mul : public Arithmetic {
...
@@ -35,6 +35,29 @@ class Mul : public Arithmetic {
#else
#else
explicit
Mul
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Mul
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Mul
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateMul
(
fbb
,
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Mul
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetActivationType
()
const
;
int
GetActivationType
()
const
;
...
...
mindspore/lite/src/ops/nchw2nhwc.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Nchw2Nhwc : public PrimitiveC {
...
@@ -33,6 +33,26 @@ class Nchw2Nhwc : public PrimitiveC {
explicit
Nchw2Nhwc
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Nchw2Nhwc
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
Nchw2Nhwc
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Nchw2Nhwc
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateNchw2Nhwc
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Nchw2Nhwc
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
};
};
...
...
mindspore/lite/src/ops/nhwc2nchw.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Nhwc2Nchw : public PrimitiveC {
...
@@ -33,6 +33,26 @@ class Nhwc2Nchw : public PrimitiveC {
explicit
Nhwc2Nchw
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Nhwc2Nchw
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
Nhwc2Nchw
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Nhwc2Nchw
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateNhwc2Nchw
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Nhwc2Nchw
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
};
};
...
...
mindspore/lite/src/ops/not_equal.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class NotEqual : public Arithmetic {
...
@@ -33,6 +33,26 @@ class NotEqual : public Arithmetic {
explicit
NotEqual
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
NotEqual
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
NotEqual
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
NotEqual
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateNotEqual
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_NotEqual
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/one_hot.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class OneHot : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class OneHot : public PrimitiveC {
#else
#else
explicit
OneHot
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
OneHot
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_OneHot
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateOneHot
(
fbb
,
attr
->
axis
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_OneHot
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetAxis
()
const
;
int
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/pad.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,34 @@ class Pad : public PrimitiveC {
...
@@ -36,6 +37,34 @@ class Pad : public PrimitiveC {
void
SetConstantValue
(
float
constant_value
);
void
SetConstantValue
(
float
constant_value
);
#else
#else
explicit
Pad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Pad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Pad
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
paddings
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
paddings
()
->
size
());
i
++
)
{
paddings
->
push_back
(
attr
->
paddings
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreatePadDirect
(
fbb
,
paddings
.
release
(),
attr
->
paddingMode
(),
attr
->
constantValue
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Pad
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetPaddings
()
const
;
std
::
vector
<
int
>
GetPaddings
()
const
;
...
...
mindspore/lite/src/ops/permute.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -32,6 +33,34 @@ class Permute : public PrimitiveC {
...
@@ -32,6 +33,34 @@ class Permute : public PrimitiveC {
explicit
Permute
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Permute
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
Permute
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Permute
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Permute
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
order
=
std
::
make_unique
<
std
::
vector
<
int64_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
order
()
->
size
());
i
++
)
{
order
->
push_back
(
attr
->
order
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreatePermuteDirect
(
fbb
,
order
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Permute
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int64_t
>
GetOrder
()
const
;
std
::
vector
<
int64_t
>
GetOrder
()
const
;
void
SetOrder
(
const
std
::
vector
<
int64_t
>
&
order
);
void
SetOrder
(
const
std
::
vector
<
int64_t
>
&
order
);
...
...
mindspore/lite/src/ops/pooling.h
浏览文件 @
02a954ab
...
@@ -47,6 +47,32 @@ class Pooling : public PrimitiveC {
...
@@ -47,6 +47,32 @@ class Pooling : public PrimitiveC {
void
SetActivationType
(
int
activation_type
);
void
SetActivationType
(
int
activation_type
);
#else
#else
explicit
Pooling
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Pooling
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Pooling
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreatePooling
(
fbb
,
attr
->
format
(),
attr
->
poolingMode
(),
attr
->
global
(),
attr
->
windowW
(),
attr
->
windowH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
roundMode
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Pooling
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetFormat
()
const
;
int
GetFormat
()
const
;
...
...
mindspore/lite/src/ops/pooling_grad.h
浏览文件 @
02a954ab
...
@@ -46,6 +46,32 @@ class PoolingGrad : public PrimitiveC {
...
@@ -46,6 +46,32 @@ class PoolingGrad : public PrimitiveC {
void
SetRoundMode
(
int
round_mode
);
void
SetRoundMode
(
int
round_mode
);
#else
#else
explicit
PoolingGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
PoolingGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_PoolingGrad
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreatePoolingGrad
(
fbb
,
attr
->
format
(),
attr
->
poolingMode
(),
attr
->
global
(),
attr
->
windowW
(),
attr
->
windowH
(),
attr
->
strideW
(),
attr
->
strideH
(),
attr
->
padMode
(),
attr
->
padUp
(),
attr
->
padDown
(),
attr
->
padLeft
(),
attr
->
padRight
(),
attr
->
roundMode
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_PoolingGrad
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetFormat
()
const
;
int
GetFormat
()
const
;
int
GetPoolingMode
()
const
;
int
GetPoolingMode
()
const
;
...
...
mindspore/lite/src/ops/power.h
浏览文件 @
02a954ab
...
@@ -36,6 +36,29 @@ class Power : public PrimitiveC {
...
@@ -36,6 +36,29 @@ class Power : public PrimitiveC {
void
SetShift
(
float
shift
);
void
SetShift
(
float
shift
);
#else
#else
explicit
Power
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Power
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Power
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreatePower
(
fbb
,
attr
->
power
(),
attr
->
scale
(),
attr
->
shift
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Power
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
float
GetPower
()
const
;
float
GetPower
()
const
;
...
...
mindspore/lite/src/ops/power_grad.h
浏览文件 @
02a954ab
...
@@ -36,6 +36,29 @@ class PowerGrad : public PrimitiveC {
...
@@ -36,6 +36,29 @@ class PowerGrad : public PrimitiveC {
void
SetShift
(
float
shift
);
void
SetShift
(
float
shift
);
#else
#else
explicit
PowerGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
PowerGrad
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_PowerGrad
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreatePowerGrad
(
fbb
,
attr
->
power
(),
attr
->
scale
(),
attr
->
shift
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_PowerGrad
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
float
GetPower
()
const
;
float
GetPower
()
const
;
float
GetScale
()
const
;
float
GetScale
()
const
;
...
...
mindspore/lite/src/ops/prelu.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/activation.h"
#include "src/ops/activation.h"
...
@@ -35,6 +36,34 @@ class Prelu : public Activation {
...
@@ -35,6 +36,34 @@ class Prelu : public Activation {
#else
#else
explicit
Prelu
(
schema
::
Primitive
*
primitive
)
:
Activation
(
primitive
)
{}
explicit
Prelu
(
schema
::
Primitive
*
primitive
)
:
Activation
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Prelu
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
slope
=
std
::
make_unique
<
std
::
vector
<
float
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
slope
()
->
size
());
i
++
)
{
slope
->
push_back
(
attr
->
slope
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreatePreluDirect
(
fbb
,
slope
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Prelu
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
float
>
GetSlope
()
const
;
std
::
vector
<
float
>
GetSlope
()
const
;
};
};
...
...
mindspore/lite/src/ops/prior_box.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -44,6 +45,47 @@ class PriorBox : public PrimitiveC {
...
@@ -44,6 +45,47 @@ class PriorBox : public PrimitiveC {
void
SetOffset
(
float
offset
);
void
SetOffset
(
float
offset
);
#else
#else
explicit
PriorBox
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
PriorBox
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_PriorBox
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
min_sizes
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
min_sizes
()
->
size
());
i
++
)
{
min_sizes
->
push_back
(
attr
->
min_sizes
()
->
data
()[
i
]);
}
auto
max_sizes
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
max_sizes
()
->
size
());
i
++
)
{
max_sizes
->
push_back
(
attr
->
max_sizes
()
->
data
()[
i
]);
}
auto
aspect_ratios
=
std
::
make_unique
<
std
::
vector
<
float
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
aspect_ratios
()
->
size
());
i
++
)
{
aspect_ratios
->
push_back
(
attr
->
aspect_ratios
()
->
data
()[
i
]);
}
auto
variances
=
std
::
make_unique
<
std
::
vector
<
float
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
variances
()
->
size
());
i
++
)
{
variances
->
push_back
(
attr
->
variances
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreatePriorBoxDirect
(
fbb
,
min_sizes
.
release
(),
max_sizes
.
release
(),
aspect_ratios
.
release
(),
variances
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_PriorBox
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetMinSizes
()
const
;
std
::
vector
<
int
>
GetMinSizes
()
const
;
...
...
mindspore/lite/src/ops/quant_dtype_cast.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class QuantDTypeCast : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class QuantDTypeCast : public PrimitiveC {
void
SetDstT
(
int
dst_t
);
void
SetDstT
(
int
dst_t
);
#else
#else
explicit
QuantDTypeCast
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
QuantDTypeCast
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_QuantDTypeCast
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateQuantDTypeCast
(
fbb
,
attr
->
srcT
(),
attr
->
dstT
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_QuantDTypeCast
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetSrcT
()
const
;
int
GetSrcT
()
const
;
...
...
mindspore/lite/src/ops/range.h
浏览文件 @
02a954ab
...
@@ -37,6 +37,29 @@ class Range : public PrimitiveC {
...
@@ -37,6 +37,29 @@ class Range : public PrimitiveC {
void
SetDelta
(
int
delta
);
void
SetDelta
(
int
delta
);
#else
#else
explicit
Range
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Range
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Range
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateRange
(
fbb
,
attr
->
dType
(),
attr
->
start
(),
attr
->
limit
(),
attr
->
delta
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Range
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetDType
()
const
;
int
GetDType
()
const
;
...
...
mindspore/lite/src/ops/rank.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Rank : public PrimitiveC {
...
@@ -33,6 +33,26 @@ class Rank : public PrimitiveC {
explicit
Rank
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Rank
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
Rank
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Rank
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateRank
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Rank
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
};
};
...
...
mindspore/lite/src/ops/reduce.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -37,6 +38,34 @@ class Reduce : public PrimitiveC {
...
@@ -37,6 +38,34 @@ class Reduce : public PrimitiveC {
void
SetMode
(
int
mode
);
void
SetMode
(
int
mode
);
#else
#else
explicit
Reduce
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Reduce
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Reduce
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axes
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axes
()
->
size
());
i
++
)
{
axes
->
push_back
(
attr
->
axes
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateReduceDirect
(
fbb
,
axes
.
release
(),
attr
->
keepDims
(),
attr
->
mode
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Reduce
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetAxes
()
const
;
std
::
vector
<
int
>
GetAxes
()
const
;
...
...
mindspore/lite/src/ops/reshape.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,34 @@ class Reshape : public PrimitiveC {
...
@@ -36,6 +37,34 @@ class Reshape : public PrimitiveC {
void
SetShape
(
const
std
::
vector
<
int64_t
>
&
shape
);
void
SetShape
(
const
std
::
vector
<
int64_t
>
&
shape
);
#else
#else
explicit
Reshape
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Reshape
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Reshape
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
shape
=
std
::
make_unique
<
std
::
vector
<
int64_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
shape
()
->
size
());
i
++
)
{
shape
->
push_back
(
attr
->
shape
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateReshapeDirect
(
fbb
,
attr
->
format
(),
shape
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Reshape
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetFormat
()
const
;
int
GetFormat
()
const
;
...
...
mindspore/lite/src/ops/resize.h
浏览文件 @
02a954ab
...
@@ -39,6 +39,31 @@ class Resize : public PrimitiveC {
...
@@ -39,6 +39,31 @@ class Resize : public PrimitiveC {
void
SetPreserveAspectRatio
(
bool
preserve_aspect_ratio
);
void
SetPreserveAspectRatio
(
bool
preserve_aspect_ratio
);
#else
#else
explicit
Resize
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Resize
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Resize
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateResize
(
fbb
,
attr
->
format
(),
attr
->
method
(),
attr
->
newHeight
(),
attr
->
newWidth
(),
attr
->
alignCorners
(),
attr
->
preserveAspectRatio
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Resize
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetFormat
()
const
;
int
GetFormat
()
const
;
...
...
mindspore/lite/src/ops/reverse.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class Reverse : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class Reverse : public PrimitiveC {
#else
#else
explicit
Reverse
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Reverse
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Reverse
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateReverseDirect
(
fbb
,
axis
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Reverse
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
};
};
...
...
mindspore/lite/src/ops/reverse_sequence.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,35 @@ class ReverseSequence : public PrimitiveC {
...
@@ -36,6 +37,35 @@ class ReverseSequence : public PrimitiveC {
void
SetSeqLengths
(
const
std
::
vector
<
int
>
&
seq_lengths
);
void
SetSeqLengths
(
const
std
::
vector
<
int
>
&
seq_lengths
);
#else
#else
explicit
ReverseSequence
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ReverseSequence
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_ReverseSequence
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
seqLengths
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
seqLengths
()
->
size
());
i
++
)
{
seqLengths
->
push_back
(
attr
->
seqLengths
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateReverseSequenceDirect
(
fbb
,
attr
->
seqAxis
(),
attr
->
batchAxis
(),
seqLengths
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ReverseSequence
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetSeqAxis
()
const
;
int
GetSeqAxis
()
const
;
...
...
mindspore/lite/src/ops/roi_pooling.h
浏览文件 @
02a954ab
...
@@ -36,6 +36,29 @@ class ROIPooling : public PrimitiveC {
...
@@ -36,6 +36,29 @@ class ROIPooling : public PrimitiveC {
void
SetScale
(
float
scale
);
void
SetScale
(
float
scale
);
#else
#else
explicit
ROIPooling
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ROIPooling
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_ROIPooling
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateROIPooling
(
fbb
,
attr
->
pooledH
(),
attr
->
pooledW
(),
attr
->
scale
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ROIPooling
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetPooledH
()
const
;
int
GetPooledH
()
const
;
...
...
mindspore/lite/src/ops/round.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Round : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Round : public ArithmeticSelf {
explicit
Round
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Round
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Round
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Round
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateRound
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Round
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/rsqrt.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Rsqrt : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Rsqrt : public ArithmeticSelf {
explicit
Rsqrt
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Rsqrt
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Rsqrt
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Rsqrt
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateRsqrt
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Rsqrt
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/scale.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Scale : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Scale : public PrimitiveC {
#else
#else
explicit
Scale
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Scale
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Scale
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateScale
(
fbb
,
attr
->
axis
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Scale
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetAxis
()
const
;
int
GetAxis
()
const
;
};
};
...
...
mindspore/lite/src/ops/scatter_nd.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class ScatterND : public PrimitiveC {
...
@@ -33,6 +33,26 @@ class ScatterND : public PrimitiveC {
explicit
ScatterND
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ScatterND
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
ScatterND
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ScatterND
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateScatterND
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ScatterND
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
};
};
...
...
mindspore/lite/src/ops/shape.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Shape : public PrimitiveC {
...
@@ -33,6 +33,26 @@ class Shape : public PrimitiveC {
explicit
Shape
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Shape
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
Shape
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Shape
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateShape
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Shape
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
};
};
...
...
mindspore/lite/src/ops/sin.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Sin : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Sin : public ArithmeticSelf {
explicit
Sin
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Sin
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Sin
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Sin
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateSin
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Sin
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/slice.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,38 @@ class Slice : public PrimitiveC {
...
@@ -36,6 +37,38 @@ class Slice : public PrimitiveC {
void
SetSize
(
const
std
::
vector
<
int
>
&
size
);
void
SetSize
(
const
std
::
vector
<
int
>
&
size
);
#else
#else
explicit
Slice
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Slice
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Slice
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
begin
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
begin
()
->
size
());
i
++
)
{
begin
->
push_back
(
attr
->
begin
()
->
data
()[
i
]);
}
auto
size
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
size
()
->
size
());
i
++
)
{
size
->
push_back
(
attr
->
size
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateSliceDirect
(
fbb
,
attr
->
format
(),
begin
.
release
(),
size
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Slice
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetFormat
()
const
;
int
GetFormat
()
const
;
...
@@ -46,6 +79,7 @@ class Slice : public PrimitiveC {
...
@@ -46,6 +79,7 @@ class Slice : public PrimitiveC {
// when running graph, we need to obtain new begins and sizes using the two function as below.
// when running graph, we need to obtain new begins and sizes using the two function as below.
std
::
vector
<
int
>
GetPostProcessBegin
()
const
;
std
::
vector
<
int
>
GetPostProcessBegin
()
const
;
std
::
vector
<
int
>
GetPostProcessSize
()
const
;
std
::
vector
<
int
>
GetPostProcessSize
()
const
;
protected:
protected:
std
::
vector
<
int
>
begin
=
{
0
};
std
::
vector
<
int
>
begin
=
{
0
};
std
::
vector
<
int
>
size
=
{
-
1
};
std
::
vector
<
int
>
size
=
{
-
1
};
...
...
mindspore/lite/src/ops/softmax.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class SoftMax : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class SoftMax : public PrimitiveC {
#else
#else
explicit
SoftMax
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
SoftMax
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_SoftMax
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateSoftMax
(
fbb
,
attr
->
axis
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_SoftMax
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetAxis
()
const
;
int
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/softmax_cross_entropy.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class SoftmaxCrossEntropy : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class SoftmaxCrossEntropy : public PrimitiveC {
#else
#else
explicit
SoftmaxCrossEntropy
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
SoftmaxCrossEntropy
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_SoftmaxCrossEntropy
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateSoftmaxCrossEntropyDirect
(
fbb
,
axis
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_SoftmaxCrossEntropy
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
};
};
...
...
mindspore/lite/src/ops/space_to_batch.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,38 @@ class SpaceToBatch : public PrimitiveC {
...
@@ -35,6 +36,38 @@ class SpaceToBatch : public PrimitiveC {
void
SetPaddings
(
const
std
::
vector
<
int
>
&
paddings
);
void
SetPaddings
(
const
std
::
vector
<
int
>
&
paddings
);
#else
#else
explicit
SpaceToBatch
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
SpaceToBatch
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_SpaceToBatch
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
blockShape
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
blockShape
()
->
size
());
i
++
)
{
blockShape
->
push_back
(
attr
->
blockShape
()
->
data
()[
i
]);
}
auto
paddings
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
paddings
()
->
size
());
i
++
)
{
paddings
->
push_back
(
attr
->
paddings
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateSpaceToBatchDirect
(
fbb
,
blockShape
.
release
(),
paddings
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_SpaceToBatch
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs
)
override
;
...
...
mindspore/lite/src/ops/space_to_batch_nd.h
浏览文件 @
02a954ab
...
@@ -18,6 +18,7 @@
...
@@ -18,6 +18,7 @@
#define LITE_MINDSPORE_LITE_C_OPS_SPACE_TO_BATCH_N_D_H_
#define LITE_MINDSPORE_LITE_C_OPS_SPACE_TO_BATCH_N_D_H_
#include <vector>
#include <vector>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -33,6 +34,38 @@ class SpaceToBatchND : public PrimitiveC {
...
@@ -33,6 +34,38 @@ class SpaceToBatchND : public PrimitiveC {
void
SetPaddings
(
const
std
::
vector
<
int
>
&
paddings
);
void
SetPaddings
(
const
std
::
vector
<
int
>
&
paddings
);
#else
#else
explicit
SpaceToBatchND
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
SpaceToBatchND
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_SpaceToBatchND
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
blockShape
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
blockShape
()
->
size
());
i
++
)
{
blockShape
->
push_back
(
attr
->
blockShape
()
->
data
()[
i
]);
}
auto
paddings
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
paddings
()
->
size
());
i
++
)
{
paddings
->
push_back
(
attr
->
paddings
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateSpaceToBatchNDDirect
(
fbb
,
blockShape
.
release
(),
paddings
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_SpaceToBatchND
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetBlockShape
()
const
;
std
::
vector
<
int
>
GetBlockShape
()
const
;
std
::
vector
<
int
>
GetPaddings
()
const
;
std
::
vector
<
int
>
GetPaddings
()
const
;
...
...
mindspore/lite/src/ops/space_to_depth.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class SpaceToDepth : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class SpaceToDepth : public PrimitiveC {
void
SetFormat
(
int
format
);
void
SetFormat
(
int
format
);
#else
#else
explicit
SpaceToDepth
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
SpaceToDepth
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_SpaceToDepth
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateSpaceToDepth
(
fbb
,
attr
->
blockSize
(),
attr
->
format
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_SpaceToDepth
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetBlockSize
()
const
;
int
GetBlockSize
()
const
;
...
...
mindspore/lite/src/ops/sparse_to_dense.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -37,6 +38,43 @@ class SparseToDense : public PrimitiveC {
...
@@ -37,6 +38,43 @@ class SparseToDense : public PrimitiveC {
void
SetValidateIndices
(
bool
validate_indices
);
void
SetValidateIndices
(
bool
validate_indices
);
#else
#else
explicit
SparseToDense
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
SparseToDense
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_SparseToDense
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
outputShape
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
outputShape
()
->
size
());
i
++
)
{
outputShape
->
push_back
(
attr
->
outputShape
()
->
data
()[
i
]);
}
auto
sparseValue
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
sparseValue
()
->
size
());
i
++
)
{
sparseValue
->
push_back
(
attr
->
sparseValue
()
->
data
()[
i
]);
}
auto
defaultValue
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
defaultValue
()
->
size
());
i
++
)
{
defaultValue
->
push_back
(
attr
->
defaultValue
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateSparseToDenseDirect
(
fbb
,
outputShape
.
release
(),
sparseValue
.
release
(),
defaultValue
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_SparseToDense
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
vector
<
int
>
GetOutputShape
()
const
;
std
::
vector
<
int
>
GetOutputShape
()
const
;
std
::
vector
<
int
>
GetSparseValue
()
const
;
std
::
vector
<
int
>
GetSparseValue
()
const
;
...
...
mindspore/lite/src/ops/split.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,34 @@ class Split : public PrimitiveC {
...
@@ -36,6 +37,34 @@ class Split : public PrimitiveC {
void
SetSplitDim
(
int
split_dim
);
void
SetSplitDim
(
int
split_dim
);
#else
#else
explicit
Split
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Split
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Split
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
sizeSplits
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
sizeSplits
()
->
size
());
i
++
)
{
sizeSplits
->
push_back
(
attr
->
sizeSplits
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateSplitDirect
(
fbb
,
attr
->
numberSplit
(),
sizeSplits
.
release
(),
attr
->
splitDim
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Split
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetNumberSplit
()
const
;
int
GetNumberSplit
()
const
;
...
...
mindspore/lite/src/ops/sqrt.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class Sqrt : public ArithmeticSelf {
...
@@ -33,6 +33,26 @@ class Sqrt : public ArithmeticSelf {
explicit
Sqrt
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Sqrt
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Sqrt
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Sqrt
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateSqrt
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Sqrt
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/square.h
浏览文件 @
02a954ab
...
@@ -32,6 +32,26 @@ class Square : public ArithmeticSelf {
...
@@ -32,6 +32,26 @@ class Square : public ArithmeticSelf {
explicit
Square
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Square
(
schema
::
PrimitiveT
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
#else
#else
explicit
Square
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
explicit
Square
(
schema
::
Primitive
*
primitive
)
:
ArithmeticSelf
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateSquare
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Square
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/squared_difference.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class SquaredDifference : public Arithmetic {
...
@@ -33,6 +33,26 @@ class SquaredDifference : public Arithmetic {
explicit
SquaredDifference
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
SquaredDifference
(
schema
::
PrimitiveT
*
primitive
)
:
Arithmetic
(
primitive
)
{}
#else
#else
explicit
SquaredDifference
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
SquaredDifference
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateSquaredDifference
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_SquaredDifference
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/squeeze.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class Squeeze : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class Squeeze : public PrimitiveC {
#else
#else
explicit
Squeeze
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Squeeze
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Squeeze
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateSqueezeDirect
(
fbb
,
axis
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Squeeze
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/stack.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,34 @@ class Stack : public PrimitiveC {
...
@@ -36,6 +37,34 @@ class Stack : public PrimitiveC {
void
SetIsScale
(
const
std
::
vector
<
int
>
&
is_scale
);
void
SetIsScale
(
const
std
::
vector
<
int
>
&
is_scale
);
#else
#else
explicit
Stack
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Stack
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Stack
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
isScale
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
isScale
()
->
size
());
i
++
)
{
isScale
->
push_back
(
attr
->
isScale
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateStackDirect
(
fbb
,
attr
->
axis
(),
attr
->
n
(),
isScale
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Stack
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetAxis
()
const
;
int
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/strided_slice.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -42,6 +43,48 @@ class StridedSlice : public PrimitiveC {
...
@@ -42,6 +43,48 @@ class StridedSlice : public PrimitiveC {
void
SetIsScale
(
const
std
::
vector
<
int
>
&
is_scale
);
void
SetIsScale
(
const
std
::
vector
<
int
>
&
is_scale
);
#else
#else
explicit
StridedSlice
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
StridedSlice
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_StridedSlice
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
begin
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
begin
()
->
size
());
i
++
)
{
begin
->
push_back
(
attr
->
begin
()
->
data
()[
i
]);
}
auto
end
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
end
()
->
size
());
i
++
)
{
end
->
push_back
(
attr
->
end
()
->
data
()[
i
]);
}
auto
stride
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
stride
()
->
size
());
i
++
)
{
stride
->
push_back
(
attr
->
stride
()
->
data
()[
i
]);
}
auto
isScale
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
isScale
()
->
size
());
i
++
)
{
isScale
->
push_back
(
attr
->
isScale
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateStridedSliceDirect
(
fbb
,
attr
->
beginMask
(),
attr
->
endMask
(),
attr
->
ellipsisMask
(),
attr
->
newAxisMask
(),
attr
->
shrinkAxisMask
(),
begin
.
release
(),
end
.
release
(),
stride
.
release
(),
isScale
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_StridedSlice
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetBeginMask
()
const
;
int
GetBeginMask
()
const
;
...
...
mindspore/lite/src/ops/sub.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Sub : public Arithmetic {
...
@@ -35,6 +35,29 @@ class Sub : public Arithmetic {
#else
#else
explicit
Sub
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
explicit
Sub
(
schema
::
Primitive
*
primitive
)
:
Arithmetic
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Sub
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateSub
(
fbb
,
attr
->
activationType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Sub
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
GetActivationType
()
const
;
int
GetActivationType
()
const
;
};
};
...
...
mindspore/lite/src/ops/tile.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,38 @@ class Tile : public PrimitiveC {
...
@@ -36,6 +37,38 @@ class Tile : public PrimitiveC {
#else
#else
explicit
Tile
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Tile
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Tile
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
multiples
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
multiples
()
->
size
());
i
++
)
{
multiples
->
push_back
(
attr
->
multiples
()
->
data
()[
i
]);
}
auto
dims
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
dims
()
->
size
());
i
++
)
{
dims
->
push_back
(
attr
->
dims
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateTileDirect
(
fbb
,
multiples
.
release
(),
dims
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Tile
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetMultiples
()
const
;
std
::
vector
<
int
>
GetMultiples
()
const
;
...
...
mindspore/lite/src/ops/topk.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class TopK : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class TopK : public PrimitiveC {
void
SetSorted
(
bool
sorted
);
void
SetSorted
(
bool
sorted
);
#else
#else
explicit
TopK
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
TopK
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_TopK
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateTopK
(
fbb
,
attr
->
k
(),
attr
->
sorted
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_TopK
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetK
()
const
;
int
GetK
()
const
;
...
...
mindspore/lite/src/ops/transpose.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,34 @@ class Transpose : public PrimitiveC {
...
@@ -36,6 +37,34 @@ class Transpose : public PrimitiveC {
void
SetConjugate
(
bool
conjugate
);
void
SetConjugate
(
bool
conjugate
);
#else
#else
explicit
Transpose
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Transpose
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Transpose
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
perm
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
perm
()
->
size
());
i
++
)
{
perm
->
push_back
(
attr
->
perm
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateTransposeDirect
(
fbb
,
perm
.
release
(),
attr
->
conjugate
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Transpose
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetPerm
()
const
;
std
::
vector
<
int
>
GetPerm
()
const
;
...
...
mindspore/lite/src/ops/tuple_get_item.h
浏览文件 @
02a954ab
...
@@ -31,6 +31,26 @@ class TupleGetItem : public PrimitiveC {
...
@@ -31,6 +31,26 @@ class TupleGetItem : public PrimitiveC {
int
UnPackAttr
(
const
Primitive
&
prim
,
const
std
::
vector
<
AnfNodePtr
>
&
inputs
);
int
UnPackAttr
(
const
Primitive
&
prim
,
const
std
::
vector
<
AnfNodePtr
>
&
inputs
);
#else
#else
explicit
TupleGetItem
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
TupleGetItem
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateTupleGetItem
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_TupleGetItem
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
};
};
}
// namespace lite
}
// namespace lite
...
...
mindspore/lite/src/ops/unique.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Unique : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Unique : public PrimitiveC {
#else
#else
explicit
Unique
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Unique
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Unique
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateUnique
(
fbb
,
attr
->
outType
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Unique
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetOutType
()
const
;
int
GetOutType
()
const
;
...
...
mindspore/lite/src/ops/unsqueeze.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class Unsqueeze : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class Unsqueeze : public PrimitiveC {
#else
#else
explicit
Unsqueeze
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Unsqueeze
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Unsqueeze
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
axis
=
std
::
make_unique
<
std
::
vector
<
int32_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
axis
()
->
size
());
i
++
)
{
axis
->
push_back
(
attr
->
axis
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateUnsqueezeDirect
(
fbb
,
axis
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Unsqueeze
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
int
>
GetAxis
()
const
;
std
::
vector
<
int
>
GetAxis
()
const
;
...
...
mindspore/lite/src/ops/unstack.h
浏览文件 @
02a954ab
...
@@ -35,6 +35,29 @@ class Unstack : public PrimitiveC {
...
@@ -35,6 +35,29 @@ class Unstack : public PrimitiveC {
void
SetAxis
(
int
axis
);
void
SetAxis
(
int
axis
);
#else
#else
explicit
Unstack
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Unstack
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Unstack
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
val_offset
=
schema
::
CreateUnstack
(
fbb
,
attr
->
num
(),
attr
->
axis
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Unstack
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
GetNum
()
const
;
int
GetNum
()
const
;
...
...
mindspore/lite/src/ops/upsample.h
浏览文件 @
02a954ab
...
@@ -21,6 +21,7 @@
...
@@ -21,6 +21,7 @@
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <string>
#include <string>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -36,6 +37,34 @@ class Upsample : public PrimitiveC {
...
@@ -36,6 +37,34 @@ class Upsample : public PrimitiveC {
void
SetScales
(
const
std
::
vector
<
float
>
&
scales
);
void
SetScales
(
const
std
::
vector
<
float
>
&
scales
);
#else
#else
explicit
Upsample
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Upsample
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Upsample
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
scales
=
std
::
make_unique
<
std
::
vector
<
float
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
scales
()
->
size
());
i
++
)
{
scales
->
push_back
(
attr
->
scales
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateUpsampleDirect
(
fbb
,
attr
->
mode
()
->
c_str
(),
scales
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Upsample
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
std
::
string
GetMode
()
const
;
std
::
string
GetMode
()
const
;
std
::
vector
<
float
>
GetScales
()
const
;
std
::
vector
<
float
>
GetScales
()
const
;
...
...
mindspore/lite/src/ops/where.h
浏览文件 @
02a954ab
...
@@ -20,6 +20,7 @@
...
@@ -20,6 +20,7 @@
#include <vector>
#include <vector>
#include <set>
#include <set>
#include <cmath>
#include <cmath>
#include <memory>
#include "ir/dtype/type_id.h"
#include "ir/dtype/type_id.h"
#include "src/ops/primitive_c.h"
#include "src/ops/primitive_c.h"
...
@@ -35,6 +36,34 @@ class Where : public PrimitiveC {
...
@@ -35,6 +36,34 @@ class Where : public PrimitiveC {
#else
#else
explicit
Where
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
Where
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
attr
=
primitive
->
value_as_Where
();
MS_ASSERT
(
attr
!=
nullptr
);
auto
condition
=
std
::
make_unique
<
std
::
vector
<
uint8_t
>>
();
for
(
int
i
=
0
;
i
<
static_cast
<
int
>
(
attr
->
condition
()
->
size
());
i
++
)
{
condition
->
push_back
(
attr
->
condition
()
->
data
()[
i
]);
}
auto
val_offset
=
schema
::
CreateWhereDirect
(
fbb
,
condition
.
release
());
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_Where
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
std
::
vector
<
bool
>
GetCondition
()
const
;
std
::
vector
<
bool
>
GetCondition
()
const
;
...
...
mindspore/lite/src/ops/zeros_like.h
浏览文件 @
02a954ab
...
@@ -33,6 +33,26 @@ class ZerosLike : public PrimitiveC {
...
@@ -33,6 +33,26 @@ class ZerosLike : public PrimitiveC {
explicit
ZerosLike
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ZerosLike
(
schema
::
PrimitiveT
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
#else
#else
explicit
ZerosLike
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
explicit
ZerosLike
(
schema
::
Primitive
*
primitive
)
:
PrimitiveC
(
primitive
)
{}
schema
::
Primitive
*
Init
(
schema
::
Primitive
*
primitive
)
{
flatbuffers
::
FlatBufferBuilder
fbb
(
1024
);
auto
val_offset
=
schema
::
CreateZerosLike
(
fbb
);
auto
prim_offset
=
schema
::
CreatePrimitive
(
fbb
,
schema
::
PrimitiveType_ZerosLike
,
val_offset
.
o
);
fbb
.
Finish
(
prim_offset
);
auto
buf
=
fbb
.
GetBufferPointer
();
MS_ASSERT
(
buf
!=
nullptr
);
auto
buf_bak
=
new
char
[
fbb
.
GetSize
()];
memcpy
(
buf_bak
,
buf
,
fbb
.
GetSize
());
auto
root
=
flatbuffers
::
GetRoot
<
schema
::
Primitive
>
(
buf_bak
);
auto
prim
=
const_cast
<
schema
::
Primitive
*>
(
root
);
delete
[]
buf_bak
;
fbb
.
Clear
();
return
prim
;
}
#endif
#endif
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
int
InferShape
(
std
::
vector
<
lite
::
tensor
::
Tensor
*>
inputs_
,
std
::
vector
<
lite
::
tensor
::
Tensor
*>
outputs_
)
override
;
};
};
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录