Searched refs:CountGetDimensionSize (Results 1 – 1 of 1) sorted by relevance
42 int64 CountGetDimensionSize(const HloModule& module) { in CountGetDimensionSize() function in xla::__anonc3569f1a0111::BatchNormExpanderTest85 EXPECT_EQ(CountGetDimensionSize(*module), 3); in TEST_F()128 EXPECT_EQ(CountGetDimensionSize(*module), 3); in TEST_F()