Home
last modified time | relevance | path

Searched refs:expectedOutputShape (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/
DUtilsTest.cpp39 Shape expectedOutputShape; in TEST() local
40 expectedOutputShape.dimensions = {4, 3, 2, 5}; in TEST()
44 EXPECT_THAT(actualOutputShape.dimensions, ElementsAreArray(expectedOutputShape.dimensions)); in TEST()
47 EXPECT_THAT(actualOutputShape.dimensions, ElementsAreArray(expectedOutputShape.dimensions)); in TEST()
/frameworks/ml/nn/runtime/test/
DTestUnspecifiedDimensions.cpp377 std::vector<uint32_t> expectedOutputShape = {valueB, kDimAGood}; in TestOne() local
378 ASSERT_EQ(outputShape, expectedOutputShape); in TestOne()