| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | costmodel_manager.cc | 39 CostModel* cost_model = new CostModel(false); in FindOrCreateCostModel() local
40 cost_model->InitFromGraph(*graph); in FindOrCreateCostModel()
41 cost_models_.emplace(graph, cost_model); in FindOrCreateCostModel()
42 return cost_model; in FindOrCreateCostModel()
64 CostModel* cost_model = it->second; in AddToCostGraphDef() local
65 cost_model->AddToCostGraphDef(graph, cost_graph); in AddToCostGraphDef()
|
| D | optimization_registry.h | 41 const CostModel* cost_model = nullptr; member
|
| /external/webp/src/enc/ |
| D | backward_references_cost_enc.c | 123 const CostModel* const cost_model, int idx, int use_color_cache, in AddSingleLiteralWithCostModel() argument
131 cost_val += GetCacheCost(cost_model, ix) * mul0; in AddSingleLiteralWithCostModel()
135 cost_val += GetLiteralCost(cost_model, color) * mul1; in AddSingleLiteralWithCostModel()
252 const CostModel* const cost_model) { in CostManagerInit() argument
266 manager->cost_cache_[0] = GetLengthCost(cost_model, 0); in CostManagerInit()
268 manager->cost_cache_[i] = GetLengthCost(cost_model, i); in CostManagerInit()
576 CostModel* const cost_model = in BackwardReferencesHashChainDistanceOnly() local
586 if (cost_model == NULL || cost_manager == NULL) goto Error; in BackwardReferencesHashChainDistanceOnly()
588 cost_model->literal_ = (double*)(cost_model + 1); in BackwardReferencesHashChainDistanceOnly()
594 if (!CostModelBuild(cost_model, xsize, cache_bits, refs)) { in BackwardReferencesHashChainDistanceOnly()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.h | 33 SlackAnalysis(const Graph* g, const CostModel* cost_model);
71 GreedyScheduler(const DeviceSet* devices, const CostModel* cost_model,
95 PriorityScheduler(const DeviceSet* devices, const CostModel* cost_model,
|
| D | scheduler.cc | 74 SlackAnalysis::SlackAnalysis(const Graph* g, const CostModel* cost_model) in SlackAnalysis() argument
75 : graph_(g), cost_model_(cost_model) {} in SlackAnalysis()
190 const CostModel* cost_model, const Graph* g, in GreedyScheduler() argument
193 cost_model_(cost_model), in GreedyScheduler()
294 const CostModel* cost_model, in PriorityScheduler() argument
296 : devices_(devices), cost_model_(cost_model), graph_(g) {} in PriorityScheduler()
|
| D | master_session.cc | 424 CostModel cost_model(true); in DoBuildPartitions() local
425 cost_model.InitFromGraph(client_graph->graph); in DoBuildPartitions()
428 SlackAnalysis sa(&client_graph->graph, &cost_model); in DoBuildPartitions()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 413 static void AddNodesToCostModel(const Graph& g, CostModel* cost_model) { in AddNodesToCostModel() argument
416 cost_model->SetNumOutputs(n, num_outputs); in AddNodesToCostModel()
419 cost_model->RecordSize(n, output, Bytes(1)); in AddNodesToCostModel()
424 static void AssignSizes(const Graph& g, CostModel* cost_model) { in AssignSizes() argument
434 cost_model->RecordSize(src, e->src_output(), size); in AssignSizes()
444 static Microseconds TimeEstimateForNode(CostModel* cost_model, Node* n) { in TimeEstimateForNode() argument
454 static void EstimateComputationCosts(const Graph& g, CostModel* cost_model) { in EstimateComputationCosts() argument
457 cost_model->RecordTime(n, TimeEstimateForNode(cost_model, n)); in EstimateComputationCosts()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter_test.cc | 112 auto cost_model = [](HloInstruction* conv) { return false; }; in TEST_F() local
113 ConvolutionGroupConverter converter(cost_model, /*convert_batch_groups_only=*/ in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 270 auto cost_model = [](HloInstruction* conv) { in RunHloPassesThroughLayoutAssn() local
275 cost_model, in RunHloPassesThroughLayoutAssn()
278 cost_model, in RunHloPassesThroughLayoutAssn()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 194 auto cost_model = [](HloInstruction* conv) { in OptimizeHloModule() local
200 cost_model, in OptimizeHloModule()
|