Home
last modified time | relevance | path

Searched refs:NumCoefficients (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes_test.cc115 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()
Dsymbolic_shapes.h41 int64 NumCoefficients(const TensorShapeProto& shape);
Dsymbolic_shapes.cc59 int64 NumCoefficients(const TensorShapeProto& shape) { in NumCoefficients() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc70 if (NumCoefficients(reduction_indices_shape) == 1) { in Optimize()
Dpin_to_host_optimizer.cc62 const int64 size = NumCoefficients(prop.shape()); in IsTensorSmall()