Home
last modified time | relevance | path

Searched defs:cost_model (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcostmodel_manager.cc39 CostModel* cost_model = new CostModel(false); in FindOrCreateCostModel() local
64 CostModel* cost_model = it->second; in AddToCostGraphDef() local
Doptimization_registry.h41 const CostModel* cost_model = nullptr; member
/external/tensorflow/tensorflow/core/distributed_runtime/
Dscheduler.cc74 SlackAnalysis::SlackAnalysis(const Graph* g, const CostModel* cost_model) in SlackAnalysis()
190 const CostModel* cost_model, const Graph* g, in GreedyScheduler()
294 const CostModel* cost_model, in PriorityScheduler()
Dmaster_session.cc424 CostModel cost_model(true); in DoBuildPartitions() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter_test.cc112 auto cost_model = [](HloInstruction* conv) { return false; }; in TEST_F() local
/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc413 static void AddNodesToCostModel(const Graph& g, CostModel* cost_model) { in AddNodesToCostModel()
424 static void AssignSizes(const Graph& g, CostModel* cost_model) { in AssignSizes()
444 static Microseconds TimeEstimateForNode(CostModel* cost_model, Node* n) { in TimeEstimateForNode()
454 static void EstimateComputationCosts(const Graph& g, CostModel* cost_model) { in EstimateComputationCosts()
/external/webp/src/enc/
Dbackward_references_cost_enc.c123 const CostModel* const cost_model, int idx, int use_color_cache, in AddSingleLiteralWithCostModel()
252 const CostModel* const cost_model) { in CostManagerInit()
576 CostModel* const cost_model = in BackwardReferencesHashChainDistanceOnly() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc194 auto cost_model = [](HloInstruction* conv) { in OptimizeHloModule() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc270 auto cost_model = [](HloInstruction* conv) { in RunHloPassesThroughLayoutAssn() local