Searched refs:DescribeBinaryOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 321 OpContext DescribeBinaryOp(const string& op, int size1, int size2) { in DescribeBinaryOp() function
866 auto cost = PredictCosts(DescribeBinaryOp("Dummy", 1000, 1)); in TEST_F()
879 auto cost = PredictCosts(DescribeBinaryOp("Dummy", 1000, 1)); in TEST_F()
1006 auto cost = PredictCosts(DescribeBinaryOp("Mul", 1000, 1)); in TEST_F()
1018 auto cost = PredictCosts(DescribeBinaryOp("Mul", 1000, 2)); in TEST_F()
1030 auto cost = PredictCosts(DescribeBinaryOp("Mod", 1000, 1)); in TEST_F()
1042 auto cost = PredictCosts(DescribeBinaryOp("SquaredDifference", 1000, 2)); in TEST_F()
1094 DescribeBinaryOp(binary_op.first, kTensorSize1, kTensorSize2); in TEST_F()
|