Searched refs:PredictAvgPool (Results 1 – 3 of 3) sorted by relevance
167 Status PredictAvgPool(const OpContext& op_context,
514 wrap(&OpLevelCostEstimator::PredictAvgPool)); in OpLevelCostEstimator()2287 Status OpLevelCostEstimator::PredictAvgPool(const OpContext& op_context, in PredictAvgPool() function in tensorflow::grappler::OpLevelCostEstimator
1473 TEST_F(OpLevelCostEstimatorTest, PredictAvgPool) { in TEST_F() argument