Searched defs:cost_model (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter_test.cc | 51 auto cost_model = [](HloInstruction* conv) { return true; }; in TEST_F() local
83 auto cost_model = [](HloInstruction* conv) { return true; }; in TEST_F() local
110 auto cost_model = [](HloInstruction* conv) { return false; }; in TEST_F() local
137 auto cost_model = [](HloInstruction* conv) { return false; }; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | costmodel_manager.cc | 39 CostModel* cost_model = new CostModel(false); in FindOrCreateCostModel() local
64 CostModel* cost_model = it->second; in AddToCostGraphDef() local
|
| D | optimization_registry.h | 42 const CostModel* cost_model = nullptr; member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 74 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()
|
| D | master_session.cc | 428 CostModel cost_model(true); in DoBuildPartitions() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 416 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/ |
| D | backward_references_cost_enc.c | 123 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/cpu/ |
| D | cpu_compiler.cc | 346 auto cost_model = [](HloInstruction* conv) { in RunHloPassesThroughLayoutAssn() local
|