Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A 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()
H A Dmaster_session.cc434 CostModel cost_model(true); in DoBuildPartitions() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcostmodel_manager.cc39 CostModel* cost_model = new CostModel(false); in FindOrCreateCostModel() local
64 CostModel* cost_model = it->second; in AddToCostGraphDef() local
H A Doptimization_registry.h42 const CostModel* cost_model = nullptr; member
/aosp_15_r20/external/webp/src/enc/
H A Dbackward_references_cost_enc.c124 const CostModel* const cost_model, int idx, int use_color_cache, in AddSingleLiteralWithCostModel()
253 const CostModel* const cost_model) { in CostManagerInit()
581 CostModel* const cost_model = in BackwardReferencesHashChainDistanceOnly() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc498 auto cost_model = [](HloInstruction* conv) { in RunHloPassesThroughLayoutAssn() local