Skip to content

Commit 7773abb

Browse files
arashadessayed
authored andcommitted
Add constexpr for FC accessing dims
1 parent e103562 commit 7773abb

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

include/mli_types.hpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,9 @@ constexpr short int kKernelONNXDWChannelInDim = kKernelONNXDWGroupDim;
6262
constexpr short int kKernelFCChannelInDim = 0;
6363
constexpr short int kKernelFCChannelOutDim = 1;
6464

65+
constexpr short int kTensorFCGroupDim = 0;
66+
constexpr short int kTensorFCChannelDim = 1;
67+
6568
constexpr short int kPerTensorQuantDim = -1;
6669
constexpr short int kSkipIterDim = -1;
6770

0 commit comments

Comments
 (0)