Searched refs:returnedAlgoCount (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cudnn_7_1.inc | 592 int *returnedAlgoCount,
597 …return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, requestedAlgoCount, returnedAlgoCount, perf…
610 int *returnedAlgoCount,
617 …handle, xDesc, x, wDesc, w, convDesc, yDesc, y, requestedAlgoCount, returnedAlgoCount, perfResults…
642 int *returnedAlgoCount,
647 …andle, srcDesc, filterDesc, convDesc, destDesc, requestedAlgoCount, returnedAlgoCount, perfResults…
738 int *returnedAlgoCount,
743 …return func_ptr(handle, xDesc, dyDesc, convDesc, dwDesc, requestedAlgoCount, returnedAlgoCount, pe…
756 int *returnedAlgoCount,
763 …dle, xDesc, x, dyDesc, y, convDesc, dwDesc, dw, requestedAlgoCount, returnedAlgoCount, perfResults…
[all …]
|
| D | cudnn_7_3.inc | 590 int *returnedAlgoCount,
595 …return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, requestedAlgoCount, returnedAlgoCount, perf…
608 int *returnedAlgoCount,
615 …handle, xDesc, x, wDesc, w, convDesc, yDesc, y, requestedAlgoCount, returnedAlgoCount, perfResults…
640 int *returnedAlgoCount,
645 …andle, srcDesc, filterDesc, convDesc, destDesc, requestedAlgoCount, returnedAlgoCount, perfResults…
736 int *returnedAlgoCount,
741 …return func_ptr(handle, xDesc, dyDesc, convDesc, dwDesc, requestedAlgoCount, returnedAlgoCount, pe…
754 int *returnedAlgoCount,
761 …dle, xDesc, x, dyDesc, y, convDesc, dwDesc, dw, requestedAlgoCount, returnedAlgoCount, perfResults…
[all …]
|
| D | cudnn_7_4.inc | 590 int *returnedAlgoCount,
595 …return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, requestedAlgoCount, returnedAlgoCount, perf…
608 int *returnedAlgoCount,
615 …handle, xDesc, x, wDesc, w, convDesc, yDesc, y, requestedAlgoCount, returnedAlgoCount, perfResults…
640 int *returnedAlgoCount,
645 …andle, srcDesc, filterDesc, convDesc, destDesc, requestedAlgoCount, returnedAlgoCount, perfResults…
736 int *returnedAlgoCount,
741 …return func_ptr(handle, xDesc, dyDesc, convDesc, dwDesc, requestedAlgoCount, returnedAlgoCount, pe…
754 int *returnedAlgoCount,
761 …dle, xDesc, x, dyDesc, y, convDesc, dwDesc, dw, requestedAlgoCount, returnedAlgoCount, perfResults…
[all …]
|
| D | cudnn_7_0.inc | 592 int *returnedAlgoCount,
597 …return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, requestedAlgoCount, returnedAlgoCount, perf…
610 int *returnedAlgoCount,
617 …handle, xDesc, x, wDesc, w, convDesc, yDesc, y, requestedAlgoCount, returnedAlgoCount, perfResults…
642 int *returnedAlgoCount,
647 …andle, srcDesc, filterDesc, convDesc, destDesc, requestedAlgoCount, returnedAlgoCount, perfResults…
738 int *returnedAlgoCount,
743 …return func_ptr(handle, xDesc, dyDesc, convDesc, dwDesc, requestedAlgoCount, returnedAlgoCount, pe…
756 int *returnedAlgoCount,
763 …dle, xDesc, x, dyDesc, y, convDesc, dwDesc, dw, requestedAlgoCount, returnedAlgoCount, perfResults…
[all …]
|
| D | cudnn_6_0.inc | 541 int *returnedAlgoCount,
546 …return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, requestedAlgoCount, returnedAlgoCount, perf…
559 int *returnedAlgoCount,
566 …handle, xDesc, x, wDesc, w, convDesc, yDesc, y, requestedAlgoCount, returnedAlgoCount, perfResults…
664 int *returnedAlgoCount,
669 …return func_ptr(handle, xDesc, dyDesc, convDesc, dwDesc, requestedAlgoCount, returnedAlgoCount, pe…
682 int *returnedAlgoCount,
689 …dle, xDesc, x, dyDesc, y, convDesc, dwDesc, dw, requestedAlgoCount, returnedAlgoCount, perfResults…
748 int *returnedAlgoCount,
753 …return func_ptr(handle, wDesc, dyDesc, convDesc, dxDesc, requestedAlgoCount, returnedAlgoCount, pe…
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2665 int returnedAlgoCount; in DoPrepareForConvolution() local
2673 /*requestAlgoCount=*/1, &returnedAlgoCount, in DoPrepareForConvolution()
2691 /*requestCount=*/1, /*returnedAlgoCount=*/&returnedAlgoCount, in DoPrepareForConvolution()
2709 /*requestAlgoCount=*/1, /*returnedAlgoCount=*/&returnedAlgoCount, in DoPrepareForConvolution()
|