Home
last modified time | relevance | path

Searched refs:getDimForPtr (Results 1 – 3 of 3) sorted by relevance

/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/tests/unittests/
H A DTensorsTest.cpp443 TEST(Tensor, getDimForPtr) { in TEST() argument
1413 EXPECT_EQ(H.getDimForPtr(0, 0), 0); in TEST()
1414 EXPECT_EQ(H.getDimForPtr(1, 0), 0); in TEST()
1415 EXPECT_EQ(H.getDimForPtr(2, 0), 0); in TEST()
1417 EXPECT_EQ(H.getDimForPtr(0, 1), 0); in TEST()
1418 EXPECT_EQ(H.getDimForPtr(1, 1), 1); in TEST()
1419 EXPECT_EQ(H.getDimForPtr(2, 1), 0); in TEST()
1421 EXPECT_EQ(H.getDimForPtr(0, 2), 1); in TEST()
1422 EXPECT_EQ(H.getDimForPtr(1, 2), 0); in TEST()
1423 EXPECT_EQ(H.getDimForPtr(2, 2), 0); in TEST()
[all …]
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/include/glow/Base/
H A DTensor.h1167 size_t getDimForPtr(size_t dim, size_t idx) const { in getDimForPtr() function
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Optimizer/GraphOptimizer/
H A DGraphOptimizer.cpp1745 dim_t channelId = filterH.getDimForPtr(0, i); in normalizeWeights()
1756 dim_t channelId = cbiasH.getDimForPtr(0, i); in normalizeWeights()