Searched refs:GetSimpleAttribute (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 1149 static port::StatusOr<T> GetSimpleAttribute(hipDevice_t device, in GetSimpleAttribute() function
1166 return GetSimpleAttribute<int>(device, hipDeviceAttributeMultiprocessorCount); in GetMultiprocessorCount()
1171 return GetSimpleAttribute<int64>( in GetMaxSharedMemoryPerCore()
1177 return GetSimpleAttribute<int64>(device, in GetMaxSharedMemoryPerBlock()
1183 return GetSimpleAttribute<int64>( in GetMaxThreadsPerMultiprocessor()
1189 return GetSimpleAttribute<int64>(device, in GetMaxThreadsPerBlock()
1195 return GetSimpleAttribute<int64>(device, in GetMaxRegistersPerBlock()
1201 return GetSimpleAttribute<int64>(device, hipDeviceAttributeWarpSize); in GetThreadsPerWarp()
1257 return GetSimpleAttribute<int>(device, attribute); in GetDeviceAttribute()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1389 static port::StatusOr<T> GetSimpleAttribute(CUdevice device, in GetSimpleAttribute() function
1405 return GetSimpleAttribute<int>(device, in GetMultiprocessorCount()
1411 return GetSimpleAttribute<int64>( in GetMaxSharedMemoryPerCore()
1417 return GetSimpleAttribute<int64>( in GetMaxSharedMemoryPerBlock()
1423 return GetSimpleAttribute<int64>( in GetMaxThreadsPerMultiprocessor()
1429 return GetSimpleAttribute<int64>(device, in GetMaxThreadsPerBlock()
1435 return GetSimpleAttribute<int64>(device, in GetMaxRegistersPerBlock()
1441 return GetSimpleAttribute<int64>(device, CU_DEVICE_ATTRIBUTE_WARP_SIZE); in GetThreadsPerWarp()
|