Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter_test.cc52 auto cost_model = [](HloInstruction* conv) { return true; }; in TEST_F() local
85 auto cost_model = [](HloInstruction* conv) { return true; }; in TEST_F() local
114 auto cost_model = [](HloInstruction* conv) { return false; }; in TEST_F() local
143 auto cost_model = [](HloInstruction* conv) { return false; }; in TEST_F() local
/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.h42 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.cc434 CostModel cost_model(true); in DoBuildPartitions() local
/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc416 static void AddNodesToCostModel(const Graph& g, CostModel* cost_model) { in AddNodesToCostModel()
427 static void AssignSizes(const Graph& g, CostModel* cost_model) { in AssignSizes()
447 static Microseconds TimeEstimateForNode(CostModel* cost_model, Node* n) { in TimeEstimateForNode()
457 static void EstimateComputationCosts(const Graph& g, CostModel* cost_model) { in EstimateComputationCosts()
/external/webp/src/enc/
Dbackward_references_cost_enc.c126 const CostModel* const cost_model, int idx, int use_color_cache, in AddSingleLiteralWithCostModel()
254 const CostModel* const cost_model) { in CostManagerInit()
584 CostModel* const cost_model = in BackwardReferencesHashChainDistanceOnly() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc498 auto cost_model = [](HloInstruction* conv) { in RunHloPassesThroughLayoutAssn() local