| 764 | } |
| 765 | |
| 766 | std::tuple<int, int> GetCcMajorMinor(Stream* stream) { |
| 767 | int cc_major, cc_minor; |
| 768 | stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major, |
| 769 | &cc_minor); |
| 770 | return std::make_tuple(cc_major, cc_minor); |
| 771 | } |
| 772 | |
| 773 | // Turns a ConvolutionDescriptor structure into a cudnn convolution handle |
| 774 | // within a scope. |
no test coverage detected