Searched refs:DescribeSparseTensorDenseMatMul (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 123 OpContext DescribeSparseTensorDenseMatMul(const int nnz_a, in DescribeSparseTensorDenseMatMul() function
877 PredictCosts(DescribeSparseTensorDenseMatMul(-1, {1, 1}, {1, 1})); in TEST_F()
884 PredictCosts(DescribeSparseTensorDenseMatMul(1, {-1, 1}, {1, 1})); in TEST_F()
891 PredictCosts(DescribeSparseTensorDenseMatMul(1, {1, -1}, {1, -1})); in TEST_F()
898 PredictCosts(DescribeSparseTensorDenseMatMul(1, {1, 1}, {-1, 1})); in TEST_F()
906 DescribeSparseTensorDenseMatMul(10, {1000, 100}, {50, 100})); in TEST_F()
916 DescribeSparseTensorDenseMatMul(10, {100000, 100}, {50, 100})); in TEST_F()
|