Skip to content

Commit

Permalink
Restore VariadicSplit CPU negative axis test (#130)
Browse files Browse the repository at this point in the history
  • Loading branch information
YangleiZouIntel committed Feb 2, 2021
1 parent 891dfb7 commit 5a244f2
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -17,12 +17,12 @@ namespace {
};

// Sum of elements numSplits = inputShapes[Axis]
const std::vector<std::vector<size_t>> numSplits = {
const std::vector<std::vector<int32_t>> numSplits = {
{1, 16, 5, 8},
{2, 19, 5, 4},
{7, 13, 2, 8},
{5, 8, 12, 5},
{4, 11, 6, 9}
{4, 11, -1, 9}
};

INSTANTIATE_TEST_CASE_P(smoke_NumSplitsCheck, VariadicSplitLayerTest,
Expand Down

0 comments on commit 5a244f2

Please sign in to comment.