Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h522 auto ptr_scalar = reinterpret_cast<float*>(ptr); in CudaAtomicAdd() local
523 return std::complex<float>(CudaAtomicAdd(ptr_scalar, value.real()), in CudaAtomicAdd()
524 CudaAtomicAdd(ptr_scalar + 1, value.imag())); in CudaAtomicAdd()
529 auto ptr_scalar = reinterpret_cast<double*>(ptr); in CudaAtomicAdd() local
530 return std::complex<double>(CudaAtomicAdd(ptr_scalar, value.real()), in CudaAtomicAdd()
531 CudaAtomicAdd(ptr_scalar + 1, value.imag())); in CudaAtomicAdd()