Searched refs:GetSimpleAttribute (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 1101 static port::StatusOr<T> GetSimpleAttribute(hipDevice_t device, in GetSimpleAttribute() function
1118 return GetSimpleAttribute<int>(device, hipDeviceAttributeMultiprocessorCount); in GetMultiprocessorCount()
1123 return GetSimpleAttribute<int64>( in GetMaxSharedMemoryPerCore()
1129 return GetSimpleAttribute<int64>(device, in GetMaxSharedMemoryPerBlock()
1135 return GetSimpleAttribute<int64>( in GetMaxThreadsPerMultiprocessor()
1141 return GetSimpleAttribute<int64>(device, in GetMaxThreadsPerBlock()
1147 return GetSimpleAttribute<int64>(device, in GetMaxRegistersPerBlock()
1153 return GetSimpleAttribute<int64>(device, hipDeviceAttributeWarpSize); in GetThreadsPerWarp()
1209 return GetSimpleAttribute<int>(device, attribute); in GetDeviceAttribute()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1400 static port::StatusOr<T> GetSimpleAttribute(CUdevice device, in GetSimpleAttribute() function
1412 return GetSimpleAttribute<int>(device, in GetMultiprocessorCount()
1418 return GetSimpleAttribute<int64>( in GetMaxSharedMemoryPerCore()
1424 return GetSimpleAttribute<int64>( in GetMaxSharedMemoryPerBlock()
1430 return GetSimpleAttribute<int64>( in GetMaxThreadsPerMultiprocessor()
1436 return GetSimpleAttribute<int64>(device, in GetMaxThreadsPerBlock()
1442 return GetSimpleAttribute<int64>(device, in GetMaxRegistersPerBlock()
1448 return GetSimpleAttribute<int64>(device, CU_DEVICE_ATTRIBUTE_WARP_SIZE); in GetThreadsPerWarp()
|