Searched refs:PredictMaxPool (Results 1 – 3 of 3) sorted by relevance
142 Costs PredictMaxPool(const OpContext& op_context) const;
275 {kMaxPool, wrap(&OpLevelCostEstimator::PredictMaxPool)}, in OpLevelCostEstimator()1484 Costs OpLevelCostEstimator::PredictMaxPool(const OpContext& op_context) const { in PredictMaxPool() function in tensorflow::grappler::OpLevelCostEstimator
1009 TEST_F(OpLevelCostEstimatorTest, PredictMaxPool) { in TEST_F() argument