Searched refs:DescribeBatchMatMul (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 592 OpContext DescribeBatchMatMul(const std::vector<int>& dims_a, in DescribeBatchMatMul() function in tensorflow::grappler::OpLevelBatchMatMulCostEstimatorTest
1173 auto cost = PredictCosts(DescribeBatchMatMul({}, {})); in TEST_P()
1179 auto cost = PredictCosts(DescribeBatchMatMul({2, 4}, {})); in TEST_P()
1185 auto cost = PredictCosts(DescribeBatchMatMul({2, 4}, {4, 2})); in TEST_P()
1191 auto cost = PredictCosts(DescribeBatchMatMul({1, 2, 4}, {1, 4, 2})); in TEST_P()
1197 auto cost = PredictCosts(DescribeBatchMatMul({2, 4}, {1, 3, 4, 2})); in TEST_P()
1207 CountBatchMatMulOperations(DescribeBatchMatMul({2, 4}, {4, 2}).op_info, in TEST_P()
1213 DescribeBatchMatMul({10, 2, 4}, {-1, 10, 4, 2}).op_info, in TEST_P()
1219 DescribeBatchMatMul({2, 10, 2, 4}, {-1, 10, 4, 2}).op_info, in TEST_P()
1225 DescribeBatchMatMul({2, 4}, {1, 3, 4, 2}).op_info, in TEST_P()
[all …]
|