MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetCcMajorMinor

Function GetCcMajorMinor

tensorflow/stream_executor/cuda/cuda_dnn.cc:766–771  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

764}
765
766std::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.

Calls 2

parentMethod · 0.45

Tested by

no test coverage detected