Home
last modified time | relevance | path

Searched refs:CostModel (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc32 void CostModel::SuppressInfrequent() { in SuppressInfrequent()
53 void CostModel::MergeFromLocal(const Graph& g, const CostModel& cm) { in MergeFromLocal()
77 void CostModel::MergeFromGlobal(const CostModel& cm) { in MergeFromGlobal()
99 void CostModel::MergeFromStats(const NodeNameToCostIdMap& map, in MergeFromStats()
125 void CostModel::Ensure(int id, int num_outputs) { in Ensure()
153 void CostModel::SetNumOutputs(const Node* node, int num_outputs) { in SetNumOutputs()
167 void CostModel::RecordCount(const Node* node, int count) { in RecordCount()
174 int32 CostModel::TotalCount(const Node* node) const { in TotalCount()
180 void CostModel::RecordSize(const Node* node, int slot, Bytes bytes) { in RecordSize()
194 Bytes CostModel::TotalBytes(const Node* node, int slot) const { in TotalBytes()
[all …]
Dcostmodel.h45 class CostModel {
49 explicit CostModel(bool is_global) : is_global_(is_global) { in CostModel() function
74 void MergeFromGlobal(const CostModel& cm);
78 void MergeFromLocal(const Graph& g, const CostModel& cm);
227 TF_DISALLOW_COPY_AND_ASSIGN(CostModel);
/external/tensorflow/tensorflow/core/distributed_runtime/
Dscheduler.h33 SlackAnalysis(const Graph* g, const CostModel* cost_model);
50 const CostModel* cost_model_;
71 GreedyScheduler(const DeviceSet* devices, const CostModel* cost_model,
85 const CostModel* cost_model_;
95 PriorityScheduler(const DeviceSet* devices, const CostModel* cost_model,
110 const CostModel* cost_model_;
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.cc366 CostModel cost_model(true); in DoBuildPartitions()
/external/tensorflow/tensorflow/core/common_runtime/
Dcostmodel_manager.cc33 CostModel* CostModelManager::FindOrCreateCostModel(const Graph* graph) { in FindOrCreateCostModel()
39 CostModel* cost_model = new CostModel(false); in FindOrCreateCostModel()
64 CostModel* cost_model = it->second; in AddToCostGraphDef()
Dcostmodel_manager.h34 typedef std::unordered_map<const Graph*, CostModel*> CostModelMap;
42 CostModel* FindOrCreateCostModel(const Graph* graph);
Ddirect_session_with_tracking_alloc_test.cc96 const CostModel* cm = (it).second; in TEST()
146 const CostModel* cm = (*cost_models.begin()).second; in TEST()
213 const CostModel* cm = (it).second; in TestHWAccelerator()
Doptimization_registry.h40 const CostModel* cost_model = nullptr;
Ddirect_session.h49 class CostModel; variable
Dstep_stats_collector.cc190 CostModel* cm = cost_model_manager->FindOrCreateCostModel(graph); in BuildCostModel()
/external/webp/src/enc/
Dbackward_references_cost_enc.c38 } CostModel; typedef
61 static int CostModelBuild(CostModel* const m, int xsize, int cache_bits, in CostModelBuild()
95 static WEBP_INLINE double GetLiteralCost(const CostModel* const m, uint32_t v) { in GetLiteralCost()
102 static WEBP_INLINE double GetCacheCost(const CostModel* const m, uint32_t idx) { in GetCacheCost()
107 static WEBP_INLINE double GetLengthCost(const CostModel* const m, in GetLengthCost()
114 static WEBP_INLINE double GetDistanceCost(const CostModel* const m, in GetDistanceCost()
123 const CostModel* const cost_model, int idx, int use_color_cache, in AddSingleLiteralWithCostModel()
252 const CostModel* const cost_model) { in CostManagerInit()
575 const size_t cost_model_size = sizeof(CostModel) + literal_array_size; in BackwardReferencesHashChainDistanceOnly()
576 CostModel* const cost_model = in BackwardReferencesHashChainDistanceOnly()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceThreadPool.h185 typedef TensorCostModel<ThreadPoolDevice> CostModel; in parallelFor() typedef
187 CostModel::numThreads(n, cost, static_cast<int>(numThreads())) == 1) { in parallelFor()
198 double block_size_f = 1.0 / CostModel::taskSize(1, cost); in parallelFor()
/external/llvm/lib/Analysis/
DCMakeLists.txt21 CostModel.cpp
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc63 TEST_F(VirtualClusterTest, CostModel) { in TEST_F() argument
Dsingle_machine_test.cc67 TEST_F(SingleMachineTest, CostModel) { in TEST_F() argument