Searched refs:PredictAvgPool (Results 1 – 3 of 3) sorted by relevance
144 Costs PredictAvgPool(const OpContext& op_context) const;
277 {kAvgPool, wrap(&OpLevelCostEstimator::PredictAvgPool)}, in OpLevelCostEstimator()1561 Costs OpLevelCostEstimator::PredictAvgPool(const OpContext& op_context) const { in PredictAvgPool() function in tensorflow::grappler::OpLevelCostEstimator
1092 TEST_F(OpLevelCostEstimatorTest, PredictAvgPool) { in TEST_F() argument