Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/
H A DUtilsTest.cpp47 Shape expectedOutputShape; in TEST() local
48 expectedOutputShape.dimensions = {4, 3, 2, 5}; in TEST()
52 EXPECT_THAT(actualOutputShape.dimensions, ElementsAreArray(expectedOutputShape.dimensions)); in TEST()
55 EXPECT_THAT(actualOutputShape.dimensions, ElementsAreArray(expectedOutputShape.dimensions)); in TEST()
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestUnspecifiedDimensions.cpp407 std::vector<uint32_t> expectedOutputShape = {valueB, kDimAGood}; in TestOne() local
408 ASSERT_EQ(outputShape, expectedOutputShape); in TestOne()