Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
T
tic
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
wenyuanbo
tic
Commits
3d7da362
Commit
3d7da362
authored
Jul 19, 2019
by
zacario-li
Committed by
Tianqi Chen
Jul 18, 2019
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix topi c++ conv2d_nchw lambda expr issue (#3570)
parent
2973f8a6
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 additions
and
1 deletions
+1
-1
topi/include/topi/nn.h
+1
-1
No files found.
topi/include/topi/nn.h
View file @
3d7da362
...
@@ -268,7 +268,7 @@ inline tvm::Tensor conv2d_nchw(const tvm::Tensor& I,
...
@@ -268,7 +268,7 @@ inline tvm::Tensor conv2d_nchw(const tvm::Tensor& I,
:
pad
(
I
,
{
tvm
::
Expr
(
0
),
tvm
::
Expr
(
0
),
pad_h
,
pad_w
});
:
pad
(
I
,
{
tvm
::
Expr
(
0
),
tvm
::
Expr
(
0
),
pad_h
,
pad_w
});
auto
l
=
[
&
](
tvm
::
Var
b
,
tvm
::
Var
o
,
tvm
::
Var
h
,
tvm
::
Var
w
)
{
auto
l
=
[
&
](
tvm
::
Var
b
,
tvm
::
Var
o
,
tvm
::
Var
h
,
tvm
::
Var
w
)
{
return
tvm
::
sum
(
return
tvm
::
sum
(
T
(
b
,
i
,
stride_h
*
h
+
kh
,
stride_w
*
w
+
kw
)
*
W
(
i
,
o
,
kh
,
kw
),
T
(
b
,
i
,
stride_h
*
h
+
kh
,
stride_w
*
w
+
kw
)
*
W
(
o
,
i
,
kh
,
kw
),
{
i
,
kh
,
kw
});
{
i
,
kh
,
kw
});
};
};
return
tvm
::
compute
(
output_shape
,
l
,
name
,
tag
);
return
tvm
::
compute
(
output_shape
,
l
,
name
,
tag
);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment