Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc34 void CostModel::SuppressInfrequent() { in SuppressInfrequent()
55 void CostModel::MergeFromLocal(const Graph& g, const CostModel& cm) { in MergeFromLocal()
79 void CostModel::MergeFromGlobal(const CostModel& cm) { in MergeFromGlobal()
101 void CostModel::MergeFromStats(const NodeNameToCostIdMap& map, in MergeFromStats()
128 void CostModel::Ensure(int id, int num_outputs) { in Ensure()
156 void CostModel::SetNumOutputs(const Node* node, int num_outputs) { in SetNumOutputs()
170 void CostModel::RecordCount(const Node* node, int count) { in RecordCount()
177 int32 CostModel::TotalCount(const Node* node) const { in TotalCount()
183 void CostModel::RecordSize(const Node* node, int slot, Bytes bytes) { in RecordSize()
197 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
82 void MergeFromGlobal(const CostModel& cm);
86 void MergeFromLocal(const Graph& g, const CostModel& cm);
235 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.cc434 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.cc103 const CostModel* cm = (it).second; in TEST()
163 const CostModel* cm = (*cost_models.begin()).second; in TEST()
222 const CostModel* cm = (it).second; in TestHWAccelerator()
Doptimization_registry.h42 const CostModel* cost_model = nullptr;
Ddirect_session.h50 class CostModel; variable
Dstep_stats_collector.cc315 CostModel* cm = cost_model_manager->FindOrCreateCostModel(graph); in BuildCostModel()
/external/webp/src/enc/
Dbackward_references_cost_enc.c39 } CostModel; typedef
62 static int CostModelBuild(CostModel* const m, int xsize, int cache_bits, in CostModelBuild()
96 static WEBP_INLINE float GetLiteralCost(const CostModel* const m, uint32_t v) { in GetLiteralCost()
103 static WEBP_INLINE float GetCacheCost(const CostModel* const m, uint32_t idx) { in GetCacheCost()
108 static WEBP_INLINE float GetLengthCost(const CostModel* const m, in GetLengthCost()
115 static WEBP_INLINE float GetDistanceCost(const CostModel* const m, in GetDistanceCost()
124 const CostModel* const cost_model, int idx, int use_color_cache, in AddSingleLiteralWithCostModel()
253 const CostModel* const cost_model) { in CostManagerInit()
580 const size_t cost_model_size = sizeof(CostModel) + literal_array_size; in BackwardReferencesHashChainDistanceOnly()
581 CostModel* const cost_model = in BackwardReferencesHashChainDistanceOnly()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceThreadPool.h93 const size_t num_threads = CostModel::numThreads(n, TensorOpCost(1.0, 1.0, 0), 4); in memcpy()
197 CostModel::numThreads(n, cost, static_cast<int>(numThreads())) == 1) { in parallelFor()
255 CostModel::numThreads(n, cost, static_cast<int>(numThreads())) == 1) { in parallelForAsync()
311 typedef TensorCostModel<ThreadPoolDevice> CostModel; typedef
344 const double block_size_f = 1.0 / CostModel::taskSize(1, cost); in CalculateParallelForBlock()
/external/llvm/lib/Analysis/
DCMakeLists.txt21 CostModel.cpp
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Analysis/
DCMakeLists.txt22 CostModel.cpp
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc64 TEST_F(VirtualClusterTest, CostModel) { in TEST_F() argument
Dsingle_machine_test.cc72 TEST_F(SingleMachineTest, CostModel) { in TEST_F() argument
/external/webp/
DChangeLog2262 72395ba9 Merge "Modify CostModel to allocate optimal memory."
2265 c2b5a039 Modify CostModel to allocate optimal memory.