Searched refs:getDimForPtr (Results 1 – 3 of 3) sorted by relevance
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/tests/unittests/ |
H A D | TensorsTest.cpp | 443 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 D | Tensor.h | 1167 size_t getDimForPtr(size_t dim, size_t idx) const { in getDimForPtr() function
|
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/lib/Optimizer/GraphOptimizer/ |
H A D | GraphOptimizer.cpp | 1745 dim_t channelId = filterH.getDimForPtr(0, i); in normalizeWeights() 1756 dim_t channelId = cbiasH.getDimForPtr(0, i); in normalizeWeights()
|