Searched refs:DescribeSparseTensorDenseMatMul (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 121 OpContext DescribeSparseTensorDenseMatMul(const int nnz_a, in DescribeSparseTensorDenseMatMul() function
1256 PredictCosts(DescribeSparseTensorDenseMatMul(-1, {1, 1}, {1, 1})); in TEST_F()
1263 PredictCosts(DescribeSparseTensorDenseMatMul(1, {-1, 1}, {1, 1})); in TEST_F()
1270 PredictCosts(DescribeSparseTensorDenseMatMul(1, {1, -1}, {1, -1})); in TEST_F()
1277 PredictCosts(DescribeSparseTensorDenseMatMul(1, {1, 1}, {-1, 1})); in TEST_F()
1285 DescribeSparseTensorDenseMatMul(10, {1000, 100}, {50, 100})); in TEST_F()
1295 DescribeSparseTensorDenseMatMul(10, {100000, 100}, {50, 100})); in TEST_F()
|