Home
last modified time | relevance | path

Searched refs:cost_model_manager (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcostmodel_test.cc87 CostModelManager cost_model_manager; in TEST() local
88 collector.BuildCostModel(&cost_model_manager, device_map); in TEST()
91 cost_model_manager.AddToCostGraphDef(graph1.get(), &cost_graph_def)); in TEST()
93 cost_model_manager.AddToCostGraphDef(graph2.get(), &cost_graph_def)); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h165 CostModelManager* cost_model_manager,
Dstep_stats_collector.cc260 CostModelManager* cost_model_manager, in BuildCostModel() argument
315 CostModel* cm = cost_model_manager->FindOrCreateCostModel(graph); in BuildCostModel()