| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 34 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()
127 void CostModel::Ensure(int id, int num_outputs) { in Ensure()
155 void CostModel::SetNumOutputs(const Node* node, int num_outputs) { in SetNumOutputs()
169 void CostModel::RecordCount(const Node* node, int count) { in RecordCount()
176 int32 CostModel::TotalCount(const Node* node) const { in TotalCount()
182 void CostModel::RecordSize(const Node* node, int slot, Bytes bytes) { in RecordSize()
196 Bytes CostModel::TotalBytes(const Node* node, int slot) const { in TotalBytes()
[all …]
|
| D | costmodel.h | 45 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/ |
| D | scheduler.h | 33 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_;
|
| 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | costmodel_manager.cc | 33 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()
|
| D | costmodel_manager.h | 34 typedef std::unordered_map<const Graph*, CostModel*> CostModelMap;
42 CostModel* FindOrCreateCostModel(const Graph* graph);
|
| D | direct_session_with_tracking_alloc_test.cc | 102 const CostModel* cm = (it).second; in TEST()
162 const CostModel* cm = (*cost_models.begin()).second; in TEST()
220 const CostModel* cm = (it).second; in TestHWAccelerator()
|
| D | optimization_registry.h | 41 const CostModel* cost_model = nullptr;
|
| D | direct_session.h | 50 class CostModel; variable
|
| D | step_stats_collector.cc | 315 CostModel* cm = cost_model_manager->FindOrCreateCostModel(graph); in BuildCostModel()
|
| /external/webp/src/enc/ |
| D | backward_references_cost_enc.c | 38 } 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/llvm-project/mlir/include/mlir/Rewrite/ |
| D | PatternApplicator.h | 34 using CostModel = function_ref<PatternBenefit(const Pattern &)>;
55 void applyCostModel(CostModel model);
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorDeviceThreadPool.h | 185 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/ |
| D | CMakeLists.txt | 21 CostModel.cpp
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Analysis/ |
| D | CMakeLists.txt | 22 CostModel.cpp
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster_test.cc | 64 TEST_F(VirtualClusterTest, CostModel) { in TEST_F() argument
|
| D | single_machine_test.cc | 73 TEST_F(SingleMachineTest, CostModel) { in TEST_F() argument
|
| /external/llvm-project/llvm/utils/gn/secondary/llvm/lib/Analysis/ |
| D | BUILD.gn | 39 "CostModel.cpp",
|
| /external/llvm-project/llvm/lib/Analysis/ |
| D | CMakeLists.txt | 38 CostModel.cpp
|
| /external/llvm-project/mlir/lib/Rewrite/ |
| D | PatternApplicator.cpp | 33 void PatternApplicator::applyCostModel(CostModel model) { in applyCostModel()
|