Searched refs:ComputeSizeRatio (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes_test.cc | 125 EXPECT_EQ(16, ComputeSizeRatio(MakeShape({32, 32}), MakeShape({32, 2}))); in TEST_F()
126 EXPECT_EQ(16, ComputeSizeRatio(MakeShape({-2, 32}), MakeShape({-2, 2}))); in TEST_F()
128 ComputeSizeRatio(MakeShape({-2, -2, 32}), MakeShape({-2, 2, -2}))); in TEST_F()
130 ComputeSizeRatio(MakeShape({-2, -2, 32}), MakeShape({-2, 2, 2}))); in TEST_F()
132 ComputeSizeRatio(MakeShape({-2, 2, 32}), MakeShape({-2, 2, -2}))); in TEST_F()
133 EXPECT_EQ(-1, ComputeSizeRatio(MakeShape({-2, -2}), MakeShape({-2, 2}))); in TEST_F()
134 EXPECT_EQ(-1, ComputeSizeRatio(MakeShape({-2, 32}), MakeShape({-2, -2}))); in TEST_F()
135 EXPECT_EQ(1, ComputeSizeRatio(MakeShape({-2, -3}), MakeShape({-3, -2}))); in TEST_F()
136 EXPECT_EQ(-1, ComputeSizeRatio(MakeShape({-1, 32}), MakeShape({-2, 2}))); in TEST_F()
137 EXPECT_EQ(-1, ComputeSizeRatio(MakeShape({-1, 32}), MakeShape({-2, 0}))); in TEST_F()
|
| D | symbolic_shapes.h | 71 int64 ComputeSizeRatio(const TensorShapeProto& numerator,
|
| D | symbolic_shapes.cc | 219 int64 ComputeSizeRatio(const TensorShapeProto& numerator, in ComputeSizeRatio() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 171 int64 result = ComputeSizeRatio(shape1, shape2); in Optimize()
|