Searched refs:SetNumWarmupSteps (Results 1 – 4 of 4) sorted by relevance
73 void SetNumWarmupSteps(int num_steps);
40 void Cluster::SetNumWarmupSteps(int num_steps) { in SetNumWarmupSteps() function in tensorflow::grappler::Cluster
143 cluster_->SetNumWarmupSteps(10); in TF_NewCluster()
133 cluster->SetNumWarmupSteps(10); in ConvertGraphDefToTensorRT()