Searched refs:DescribeMatMul (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 86 OpContext DescribeMatMul(int m, int n, int l, int k) { in DescribeMatMul() function
1138 auto cost = PredictCosts(DescribeMatMul(2, 4, 7, 7)); in TEST_F()
1144 auto cost = PredictCosts(DescribeMatMul(-1, 4, 7, 7)); in TEST_F()
1150 auto cost = PredictCosts(DescribeMatMul(2, 4, -1, 7)); in TEST_F()
1205 CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_P()
1210 EXPECT_EQ(10 * CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_P()
1216 EXPECT_EQ(20 * CountMatMulOperations(DescribeMatMul(2, 2, 4, 4).op_info, in TEST_P()
|