Skip to content

Commit 81e0cfa

Browse files
committed
feat(aten::squeeze|aten::unsqueeze): adding BUILD files for new squeeze
unsqueeze ops Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent e4dac2c commit 81e0cfa

File tree

4 files changed

+14
-2
lines changed

4 files changed

+14
-2
lines changed

core/conversion/converters/BUILD

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,10 @@ cc_library(
4848
"impl/unary.cpp",
4949
"impl/interpolate.cpp",
5050
"impl/select.cpp",
51+
"impl/squeeze.cpp"
5152
"impl/stack.cpp",
52-
"impl/lstm_cell.cpp"
53+
"impl/lstm_cell.cpp",
54+
"impl/unsqueeze.cpp",
5355
],
5456
deps = [
5557
"@tensorrt//:nvinfer",

tests/core/conversion/converters/BUILD

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,14 @@ converter_test(
7171
name = "test_lstm_cell"
7272
)
7373

74+
converter_test(
75+
name = "test_unsqueeze"
76+
)
77+
78+
converter_test(
79+
name = "test_squeeze"
80+
)
81+
7482
test_suite(
7583
name = "converter_tests",
7684
tests = [
@@ -88,6 +96,8 @@ test_suite(
8896
":test_interpolate",
8997
":test_select",
9098
":test_stack",
91-
":test_lstm_cell"
99+
":test_lstm_cell",
100+
":test_unsqueeze",
101+
":test_squeeze"
92102
]
93103
)

0 commit comments

Comments
 (0)