Home
last modified time | relevance | path

Searched refs:BuildCostModel (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcostmodel_test.cc88 collector.BuildCostModel(&cost_model_manager, device_map); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.h178 void BuildCostModel(Item* item, StepStatsCollector* collector,
Dgraph_mgr.cc526 BuildCostModel(item, collector, cost_graph); in StartParallelExecutors()
561 void GraphMgr::BuildCostModel(Item* item, StepStatsCollector* collector, in BuildCostModel() function in tensorflow::GraphMgr
571 collector->BuildCostModel(&cost_model_manager_, device_to_graph); in BuildCostModel()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h164 void BuildCostModel(
Dstep_stats_collector.cc259 void StepStatsCollector::BuildCostModel( in BuildCostModel() function in tensorflow::StepStatsCollector
Ddirect_session.cc788 run_state.collector->BuildCostModel(&cost_model_manager_, device_to_graph); in RunInternal()