Searched refs:CalculateInputTensorSize (Results 1 – 2 of 2) sorted by relevance
274 static std::vector<int64> CalculateInputTensorSize(
1561 std::vector<int64> OpLevelCostEstimator::CalculateInputTensorSize( in CalculateInputTensorSize() function in tensorflow::grappler::OpLevelCostEstimator1634 CalculateInputTensorSize(op_info, found_unknown_shapes); in PredictDefaultNodeCosts()