Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcostmodel_manager.cc56 Status CostModelManager::AddToCostGraphDef(const Graph* graph, in AddToCostGraphDef() function in tensorflow::CostModelManager
65 cost_model->AddToCostGraphDef(graph, cost_graph); in AddToCostGraphDef()
H A Dcostmodel_manager.h46 Status AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph);
H A Ddirect_session.cc811 cost_model_manager_.AddToCostGraphDef(item.graph.get(), cost_graph)); in RunInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A 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()
H A Dcostmodel.h177 void AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph) const;
H A Dcostmodel.cc481 void CostModel::AddToCostGraphDef(const Graph* graph, in AddToCostGraphDef() function in tensorflow::CostModel
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dgraph_mgr.cc584 cost_model_manager_.AddToCostGraphDef(unit.graph.get(), cost_graph) in BuildCostModel()