Home
last modified time | relevance | path

Searched defs:BenchmarkCache (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConv_v7.cpp113 struct BenchmarkCache { struct
114 std::mutex mutex;
120 map;
122 bool find(const ConvolutionParams& params, T* results) { in find()
132 void insert(const ConvolutionParams& params, const T& results) { in insert()
H A DConv_v8.cpp286 struct BenchmarkCache { struct
287 std::list<KeyType> engine_cache_order;
294 engine_cache;
299 cudnn_frontend::ExecutionPlan* find(const KeyType& key) { in find()
316 void update(const KeyType& key, T& results) { in update()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DConv_miopen.cpp264 struct BenchmarkCache { struct
265 std::mutex mutex;
266 std::unordered_map<ConvolutionParams, T, ParamsHash, ParamsEqual> map;
268 bool find(const ConvolutionParams& params, T* results) { in find()
278 void insert(const ConvolutionParams& params, const T& results) { in insert()