Home
last modified time | relevance | path

Searched refs:compute_capability_major (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dkernel_spec.cc121 const char *CudaPtxInMemory::text(int compute_capability_major, in text() argument
123 std::tuple<int, int> capability{compute_capability_major, in text()
146 const char *CudaPtxInMemory::original_text(int compute_capability_major, in original_text() argument
148 std::tuple<int, int> capability{compute_capability_major, in original_text()
Dkernel_spec.h186 const char *text(int compute_capability_major,
191 const char *original_text(int compute_capability_major,
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.cc533 auto compute_capability_major = GetDeviceAttribute( in GetDeviceCapabilities() local
535 if (compute_capability_major) { in GetDeviceCapabilities()
539 *compute_capability_major); in GetDeviceCapabilities()
568 if (core_count && compute_capability_major && compute_capability_minor && in GetDeviceCapabilities()
572 device_properties_.computeMajor = *compute_capability_major; in GetDeviceCapabilities()