Home
last modified time | relevance | path

Searched refs:algoPerf (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/aten/src/ATen/native/cudnn/
DConv_v7.cpp549 perf_t algoPerf; in try_all() local
550 if (!only_use_default && cache.find(args.params, &algoPerf)) { in try_all()
552 f(algoPerf); in try_all()
562 for (auto& algoPerf : perfResults) { in try_all() local
564 f(algoPerf); in try_all()
565 cache.insert(args.params, algoPerf); in try_all()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcudnn_7_3.inc2244 cudnnAlgorithmPerformance_t *algoPerf, int numberToCreate) {
2249 return func_ptr(algoPerf, numberToCreate);
2253 cudnnAlgorithmPerformance_t algoPerf, cudnnAlgorithmDescriptor_t algoDesc,
2260 return func_ptr(algoPerf, algoDesc, status, time, memory);
2264 const cudnnAlgorithmPerformance_t algoPerf,
2272 return func_ptr(algoPerf, algoDesc, status, time, memory);
2276 cudnnAlgorithmPerformance_t *algoPerf, int numberToDestroy) {
2282 return func_ptr(algoPerf, numberToDestroy);
Dcudnn_7_1.inc2242 cudnnAlgorithmPerformance_t *algoPerf, int numberToCreate) {
2247 return func_ptr(algoPerf, numberToCreate);
2251 cudnnAlgorithmPerformance_t algoPerf, cudnnAlgorithmDescriptor_t algoDesc,
2258 return func_ptr(algoPerf, algoDesc, status, time, memory);
2262 const cudnnAlgorithmPerformance_t algoPerf,
2270 return func_ptr(algoPerf, algoDesc, status, time, memory);
2274 cudnnAlgorithmPerformance_t *algoPerf, int numberToDestroy) {
2280 return func_ptr(algoPerf, numberToDestroy);
Dcudnn_7_4.inc2385 cudnnAlgorithmPerformance_t *algoPerf, int numberToCreate) {
2390 return func_ptr(algoPerf, numberToCreate);
2394 cudnnAlgorithmPerformance_t algoPerf, cudnnAlgorithmDescriptor_t algoDesc,
2401 return func_ptr(algoPerf, algoDesc, status, time, memory);
2405 const cudnnAlgorithmPerformance_t algoPerf,
2413 return func_ptr(algoPerf, algoDesc, status, time, memory);
2417 cudnnAlgorithmPerformance_t *algoPerf, int numberToDestroy) {
2423 return func_ptr(algoPerf, numberToDestroy);
Dcudnn_7_6.inc3013 cudnnAlgorithmPerformance_t *algoPerf, int numberToCreate) {
3018 return func_ptr(algoPerf, numberToCreate);
3022 cudnnAlgorithmPerformance_t algoPerf, cudnnAlgorithmDescriptor_t algoDesc,
3029 return func_ptr(algoPerf, algoDesc, status, time, memory);
3033 const cudnnAlgorithmPerformance_t algoPerf,
3041 return func_ptr(algoPerf, algoDesc, status, time, memory);
3045 cudnnAlgorithmPerformance_t *algoPerf, int numberToDestroy) {
3051 return func_ptr(algoPerf, numberToDestroy);
Dcudnn_8_0.inc981 cudnnAlgorithmPerformance_t *algoPerf, int numberToCreate) {
986 return func_ptr(algoPerf, numberToCreate);
990 cudnnAlgorithmPerformance_t *algoPerf, int numberToDestroy) {
996 return func_ptr(algoPerf, numberToDestroy);