Searched refs:NumCoefficients (Results 1 – 5 of 5) sorted by relevance
115 EXPECT_EQ(32 * 32, NumCoefficients(MakeShape({32, 32}))); in TEST_F()117 EXPECT_EQ(-1, NumCoefficients(MakeShape({-2, 32}))); in TEST_F()121 EXPECT_EQ(-1, NumCoefficients(shape)); in TEST_F()
41 int64 NumCoefficients(const TensorShapeProto& shape);
62 int64 NumCoefficients(const TensorShapeProto& shape) { in NumCoefficients() function
108 if (NumCoefficients(reduction_indices_shape) == 1) { in Optimize()
64 const int64 size = NumCoefficients(prop.shape()); in IsTensorSmall()