Searched refs:PredictConv2D (Results 1 – 2 of 2) sorted by relevance
129 Costs PredictConv2D(const OpContext& op_context) const;
230 {kConv2d, wrap(&OpLevelCostEstimator::PredictConv2D)}, in OpLevelCostEstimator()240 {kDepthwiseConv2dNative, wrap(&OpLevelCostEstimator::PredictConv2D)}, in OpLevelCostEstimator()1100 Costs OpLevelCostEstimator::PredictConv2D(const OpContext& op_context) const { in PredictConv2D() function in tensorflow::grappler::OpLevelCostEstimator