Searched refs:cost_measure (Results 1 – 1 of 1) sorted by relevance
173 tensorflow::grappler::CostEstimator* cost_measure, in _GetOpPerformanceDataAndRunTime() argument176 tensorflow::Status status = cost_measure->Initialize(item); in _GetOpPerformanceDataAndRunTime()181 cost_measure->PredictCosts(item.graph, &run_metadata, costs)); in _GetOpPerformanceDataAndRunTime()324 tensorflow::grappler::MeasuringCostEstimator cost_measure(cluster.get(), num_measurements, 0); in TF_MeasureCosts()328 *item, &cost_measure, &op_performance_data, &costs); in TF_MeasureCosts()