Home
last modified time | relevance | path

Searched refs:AddToCostGraphDef (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcostmodel_manager.cc56 Status CostModelManager::AddToCostGraphDef(const Graph* graph, in AddToCostGraphDef() function in tensorflow::CostModelManager
65 cost_model->AddToCostGraphDef(graph, cost_graph); in AddToCostGraphDef()
Dcostmodel_manager.h46 Status AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph);
Ddirect_session.cc794 cost_model_manager_.AddToCostGraphDef(item.graph.get(), cost_graph)); in RunInternal()
/external/tensorflow/tensorflow/core/graph/
Dcostmodel_test.cc91 cost_model_manager.AddToCostGraphDef(graph1.get(), &cost_graph_def)); in TEST()
93 cost_model_manager.AddToCostGraphDef(graph2.get(), &cost_graph_def)); in TEST()
Dcostmodel.h177 void AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph) const;
Dcostmodel.cc480 void CostModel::AddToCostGraphDef(const Graph* graph, in AddToCostGraphDef() function in tensorflow::CostModel
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc575 cost_model_manager_.AddToCostGraphDef(unit.graph.get(), cost_graph) in BuildCostModel()