Searched refs:KernelMap (Results 1 – 5 of 5) sorted by relevance
| /external/swiftshader/third_party/llvm-16.0/llvm/lib/BinaryFormat/ |
| D | AMDGPUMetadataVerifier.cpp | 198 auto &KernelMap = Node.getMap(); in verifyKernel() local
200 if (!verifyScalarEntry(KernelMap, ".name", true, in verifyKernel()
203 if (!verifyScalarEntry(KernelMap, ".symbol", true, in verifyKernel()
206 if (!verifyScalarEntry(KernelMap, ".language", false, in verifyKernel()
220 KernelMap, ".language_version", false, [this](msgpack::DocNode &Node) { in verifyKernel()
226 if (!verifyEntry(KernelMap, ".args", false, [this](msgpack::DocNode &Node) { in verifyKernel()
232 if (!verifyEntry(KernelMap, ".reqd_workgroup_size", false, in verifyKernel()
241 if (!verifyEntry(KernelMap, ".workgroup_size_hint", false, in verifyKernel()
250 if (!verifyScalarEntry(KernelMap, ".vec_type_hint", false, in verifyKernel()
253 if (!verifyScalarEntry(KernelMap, ".device_enqueue_symbol", false, in verifyKernel()
[all …]
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/BinaryFormat/ |
| D | AMDGPUMetadataVerifier.cpp | 202 auto &KernelMap = Node.getMap(); in verifyKernel() local
204 if (!verifyScalarEntry(KernelMap, ".name", true, in verifyKernel()
207 if (!verifyScalarEntry(KernelMap, ".symbol", true, in verifyKernel()
210 if (!verifyScalarEntry(KernelMap, ".language", false, in verifyKernel()
224 KernelMap, ".language_version", false, [this](msgpack::DocNode &Node) { in verifyKernel()
230 if (!verifyEntry(KernelMap, ".args", false, [this](msgpack::DocNode &Node) { in verifyKernel()
236 if (!verifyEntry(KernelMap, ".reqd_workgroup_size", false, in verifyKernel()
245 if (!verifyEntry(KernelMap, ".workgroup_size_hint", false, in verifyKernel()
254 if (!verifyScalarEntry(KernelMap, ".vec_type_hint", false, in verifyKernel()
257 if (!verifyScalarEntry(KernelMap, ".device_enqueue_symbol", false, in verifyKernel()
[all …]
|
| /external/pytorch/aten/src/ATen/cuda/tunable/ |
| D | Tunable.h | 88 typedef std::unordered_map<std::string, ResultEntry> KernelMap; typedef
89 typedef std::unordered_map<std::string, KernelMap> ResultsMap;
104 KernelMap Lookup(const std::string& op_signature);
111 KernelMap& kernel_map);
121 const KernelMap& kernel_map,
128 void DisjointMerge(const std::string& op_signature, const KernelMap& kernel_map);
|
| D | Tunable.cpp | 59 KernelMap TuningResultsManager::Lookup(const std::string& op_signature) { in Lookup()
89 KernelMap& kernel_map) { in AddImpl()
134 const KernelMap& kernel_map, in DisjointMergeImpl()
135 /*out*/ std::unordered_map<std::string, KernelMap>& results) { in DisjointMergeImpl()
150 void TuningResultsManager::Load(const std::unordered_map<std::string, KernelMap>& results_to_load) { in Load()
163 void TuningResultsManager::DisjointMerge(const std::string& op_signature, const KernelMap& kernel_m… in DisjointMerge()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_segment.h | 70 typedef std::unordered_map<string, OpKernel*> KernelMap; typedef
73 KernelMap name_kernel; // op name -> kernel.
|