Searched refs:DescribeXlaEinsum (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 137 OpContext DescribeXlaEinsum(const std::vector<int>& dims_a, in DescribeXlaEinsum() function
157 OpContext op_context = DescribeXlaEinsum(dims_a, dims_b, equation); in DescribeEinsum()
1784 PredictCosts(DescribeXlaEinsum({100, 50}, {100, 50}, "ik,jk->ij")) in TEST_F()
1802 PredictCosts(DescribeXlaEinsum({25, 100, 50}, {100, 50, 25}, in TEST_F()
1823 PredictCosts(DescribeXlaEinsum({25, 16, 100, 50}, {16, 100, 50, 25}, in TEST_F()
1842 PredictCosts(DescribeXlaEinsum({25, 100, 50}, {100, 50}, "Aik,jk->Aij")) in TEST_F()
1860 PredictCosts(DescribeXlaEinsum({100, 50}, {25, 100, 50}, "ik,Bjk->ijB")) in TEST_F()
1878 PredictCosts(DescribeXlaEinsum({100, 50, 25}, {100, 50, 25}, in TEST_F()
1894 PredictCosts(DescribeXlaEinsum({100, 50}, {}, "ij->ji")) in TEST_F()
1911 PredictCosts(DescribeXlaEinsum({100, 50, 25}, {50, 100}, "ik,kl->il")) in TEST_F()
[all …]
|