Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2378 typedef typename HalfAsFloat<typename GpuComplexT<T>::type>::type CUDA_T; in DoBlasInternalImpl() typedef
2380 const size_t size = batch_count * sizeof(CUDA_T *); in DoBlasInternalImpl()
2383 DeviceMemory<CUDA_T *> a; in DoBlasInternalImpl()
2384 DeviceMemory<CUDA_T *> b; in DoBlasInternalImpl()
2385 DeviceMemory<CUDA_T *> c; in DoBlasInternalImpl()
2394 std::unique_ptr<TemporaryDeviceMemory<CUDA_T *>> a_temporary; in DoBlasInternalImpl()
2395 std::unique_ptr<TemporaryDeviceMemory<CUDA_T *>> b_temporary; in DoBlasInternalImpl()
2396 std::unique_ptr<TemporaryDeviceMemory<CUDA_T *>> c_temporary; in DoBlasInternalImpl()
2407 a = DeviceMemory<CUDA_T *>(a_bytes); in DoBlasInternalImpl()
2408 b = DeviceMemory<CUDA_T *>(b_bytes); in DoBlasInternalImpl()
[all …]