Searched refs:predict_avg_pool_grad (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 1134 auto predict_avg_pool_grad = [this](const int n, const int in, const int c, in TEST_F() local
1145 auto costs = predict_avg_pool_grad(10, 20, 384, 3, 2, "SAME"); in TEST_F()
1155 auto costs = predict_avg_pool_grad(10, 20, 384, 1, 2, "SAME"); in TEST_F()
1165 auto costs = predict_avg_pool_grad(10, 20, 384, 2, 3, "VALID"); in TEST_F()
|