Searched defs:compute_capability (Results 1 – 8 of 8) sorted by relevance
49 se::CudaComputeCapability compute_capability) in CudnnVectorizeConvolutions()
34 explicit CudnnPadForConvolutions(se::CudaComputeCapability compute_capability) in CudnnPadForConvolutions()
37 StatusOr<bool> Run(std::pair<int, int> compute_capability, in Run() argument
29 CudaComputeCapability compute_capability = 5; field
25 def device_details(device_name, compute_capability=None): argument
86 ComputeCapability compute_capability = 4; field
90 static string GetSmName(se::CudaComputeCapability compute_capability) { in GetSmName()361 llvm::Triple target_triple, se::CudaComputeCapability compute_capability, in NVPTXGetTargetMachine()534 auto compute_capability = in CompileToPtx() local
63 double compute_capability = 0.0; in GetNumGPUs() local