Searched refs:CalculateOutputTensorSize (Results 1 – 2 of 2) sorted by relevance
283 static std::vector<int64> CalculateOutputTensorSize(
1608 std::vector<int64> OpLevelCostEstimator::CalculateOutputTensorSize( in CalculateOutputTensorSize() function in tensorflow::grappler::OpLevelCostEstimator1636 CalculateOutputTensorSize(op_info, found_unknown_shapes); in PredictDefaultNodeCosts()