Searched refs:CountConv2DOperations (Results 1 – 2 of 2) sorted by relevance
230 static int64 CountConv2DOperations(const OpInfo& op_info,232 static int64 CountConv2DOperations(const OpInfo& op_info,
882 int64 OpLevelCostEstimator::CountConv2DOperations(const OpInfo& op_info, in CountConv2DOperations() function in tensorflow::grappler::OpLevelCostEstimator884 return CountConv2DOperations(op_info, nullptr, found_unknown_shapes); in CountConv2DOperations()994 int64 OpLevelCostEstimator::CountConv2DOperations( in CountConv2DOperations() function in tensorflow::grappler::OpLevelCostEstimator1669 int64 num_compute_ops = CountConv2DOperations(op_info, &found_unknown_shapes); in PredictConv2D()