Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
1e4fc4cf
M
mindspore
项目概览
magicwindyyd
/
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看板
提交
1e4fc4cf
编写于
8月 30, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
8月 30, 2020
浏览文件
操作
浏览文件
下载
差异文件
!5485 Fix a tiny mistake of split ops
Merge pull request !5485 from liuwenhao/lite
上级
bc7d0035
24b57c22
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
18 addition
and
14 deletion
+18
-14
build.bat
build.bat
+13
-13
mindspore/lite/src/runtime/kernel/arm/base/split_base.cc
mindspore/lite/src/runtime/kernel/arm/base/split_base.cc
+5
-1
未找到文件。
build.bat
浏览文件 @
1e4fc4cf
...
...
@@ -20,14 +20,14 @@ IF NOT EXIST "%BASEPATH%/build" (
md
"build"
)
cd
"
%BASEPATH%
/build"
cd
%BASEPATH%
/build
set
BUILD_PATH
=
%CD%
IF
NOT
EXIST
"
%BUILD_PATH%
/mindspore"
(
md
"mindspore"
)
cd
"
%CD%
/mindspore"
cd
%CD%
/mindspore
IF
"
%
1
%
"
==
"lite"
(
call
:gene
_gtest
...
...
@@ -56,7 +56,7 @@ IF "%1%" == "lite" (
echo
"build fail."
goto
run_fail
)
ELSE
(
cd
"
%BASEPATH%
/output"
cd
%BASEPATH%
/output
rd
/s /q
_CPack_Packages
)
)
ELSE
(
...
...
@@ -78,22 +78,22 @@ IF "%1%" == "lite" (
)
)
cd
"
%BASEPATH%
"
cd
%BASEPATH%
goto
run_eof
:run
_cmake
cd
"
%BUILD_PATH%
/mindspore"
cd
%BUILD_PATH%
/mindspore
cmake
-DBUILD
_DEVICE
=
on
-DBUILD
_CONVERTER
=
on
-DPLATFORM
_ARM64
=
off
-DSUPPORT
_TRAIN
=
off
^
-DCMAKE
_BUILD_TYPE
=
Release
-DSUPPORT
_GPU
=
off
-DBUILD
_MINDDATA
=
off
-DOFFLINE
_COMPILE
=
off
^
-G
"CodeBlocks - MinGW Makefiles"
"
%BASEPATH%
/mindspore/lite"
GOTO
:EOF
:gene
_gtest
cd
"
%BASEPATH%
/third_party"
cd
%BASEPATH%
/third
_party
IF
EXIST
googletest
rd
/s /q
googletest
git
submodule
update
--init --recursive
googletest
cd
"
%BUILD_PATH%
/mindspore"
cd
%BUILD_PATH%
/mindspore
GOTO
:EOF
:gene
_protobuf
...
...
@@ -111,22 +111,22 @@ GOTO:EOF
:gene
_flatbuffer
SET
FLATC
=
"
%BASEPATH%
/build/mindspore/_deps/flatbuffers-src/_build/flatc"
SET
FLAT_DIR
=
"
%BASEPATH%
/mindspore/lite/schema"
SET
FLAT_DIR
=
%BASEPATH%
/mindspore/lite/schema
cd
%FLAT_DIR%
IF
EXIST
inner
rd
/s /q
inner
md
inner
%FLATC%
-c -b
*
.fbs
%FLATC%
-c -b --reflect-types --gen-mutable --reflect-names --gen-object-api -o
%FLAT_DIR%
/inner
*
.fbs
%FLATC%
-c -b --reflect-types --gen-mutable --reflect-names --gen-object-api -o
"
%FLAT_DIR%
/inner"
*
.fbs
SET
FLAT_DIR
=
"
%BASEPATH%
/mindspore/lite/tools/converter/parser/tflite"
SET
FLAT_DIR
=
%BASEPATH%
/mindspore/lite/tools/converter/parser/tflite
cd
%FLAT_DIR%
%FLATC%
-c -b --reflect-types --gen-mutable --reflect-names --gen-object-api -o
%FLAT_DIR%
*
.fbs
cd
"
%BUILD_PATH%
/mindspore"
%FLATC%
-c -b --reflect-types --gen-mutable --reflect-names --gen-object-api -o
"
%FLAT_DIR%
"
*
.fbs
cd
%BUILD_PATH%
/mindspore
GOTO
:EOF
:run
_fail
cd
"
%BASEPATH%
"
cd
%BASEPATH%
set
errorlevel
=
1
:run
_eof
mindspore/lite/src/runtime/kernel/arm/base/split_base.cc
浏览文件 @
1e4fc4cf
...
...
@@ -28,7 +28,11 @@ using mindspore::lite::RET_OK;
using
mindspore
::
schema
::
PrimitiveType_Split
;
namespace
mindspore
::
kernel
{
int
SplitBaseCPUKernel
::
Init
()
{
return
RET_OK
;
}
int
SplitBaseCPUKernel
::
Init
()
{
auto
split_dim
=
param
->
split_dim_
;
param
->
split_dim_
=
split_dim
>=
0
?
split_dim
:
in_tensors_
.
front
()
->
shape
().
size
()
+
split_dim
;
return
RET_OK
;
}
int
SplitBaseCPUKernel
::
ReSize
()
{
auto
in_tensor
=
in_tensors_
.
front
();
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录