Home
last modified time | relevance | path

Searched refs:GpuLdg (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper.h130 __host__ __device__ inline tensorflow::bfloat16 GpuLdg( in GpuLdg() function
133 return_value.value = GpuLdg(reinterpret_cast<const uint16_t*>(address)); in GpuLdg()
140 return GpuLdg(ptr); in ldg()
Dgpu_device_functions.h523 __host__ __device__ T GpuLdg(const T* address) { in GpuLdg() function
531 __host__ __device__ inline bool GpuLdg(const bool* address) { in GpuLdg() function
532 return GpuLdg(reinterpret_cast<const char*>(address)) != 0; in GpuLdg()
535 __host__ __device__ inline std::complex<float> GpuLdg( in GpuLdg() function
545 __host__ __device__ inline std::complex<double> GpuLdg( in GpuLdg() function
554 CREATE_CUDA_DEVICE_FUNCTION_ALIAS(GpuLdg, CudaLdg);