Searched refs:CalculateInputSize (Results 1 – 2 of 2) sorted by relevance
270 static int64 CalculateInputSize(const OpInfo& op_info,
817 const double input_size = CalculateInputSize(op_info, &unknown_shapes); in PredictOpCountBasedCost()1549 int64 OpLevelCostEstimator::CalculateInputSize(const OpInfo& op_info, in CalculateInputSize() function in tensorflow::grappler::OpLevelCostEstimator2274 const int64 input_size = CalculateInputSize(op_info, &found_unknown_shapes); in PredictAssignVariableOps()