Searched refs:AddNodeForTest (Results 1 – 3 of 3) sorted by relevance
31 stats_->AddNodeForTest( in TFProfAdvisorTest()33 stats_->AddNodeForTest(0, CreateNode("n2", "Conv2D", {}, 0, 20, 2)); in TFProfAdvisorTest()
98 void AddNodeForTest(int64 step, std::unique_ptr<TFGraphNode> node);
371 void TFStats::AddNodeForTest(int64 step, std::unique_ptr<TFGraphNode> node) { in AddNodeForTest() function in tensorflow::tfprof::TFStats