Searched refs:predict_fused_bn (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 1176 auto predict_fused_bn = [this](const int n, const int in, const int c, in TEST_F() local
1184 auto costs = predict_fused_bn(10, 20, 96, /*is_training=*/true); in TEST_F()
1194 auto costs = predict_fused_bn(10, 20, 32, /*is_training=*/true); in TEST_F()
1204 auto costs = predict_fused_bn(10, 20, 96, /*is_training=*/false); in TEST_F()
1214 auto costs = predict_fused_bn(10, 20, 32, /*is_training=*/false); in TEST_F()
|