| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 354 virtual bool DoBlasScal(Stream *stream, uint64 elem_count, float alpha,
356 virtual bool DoBlasScal(Stream *stream, uint64 elem_count, double alpha,
358 virtual bool DoBlasScal(Stream *stream, uint64 elem_count, float alpha,
360 virtual bool DoBlasScal(Stream *stream, uint64 elem_count, double alpha,
362 virtual bool DoBlasScal(Stream *stream, uint64 elem_count,
365 virtual bool DoBlasScal(Stream *stream, uint64 elem_count,
1511 bool DoBlasScal(Stream *stream, uint64 elem_count, float alpha, \
1513 bool DoBlasScal(Stream *stream, uint64 elem_count, double alpha, \
1515 bool DoBlasScal(Stream *stream, uint64 elem_count, float alpha, \
1517 bool DoBlasScal(Stream *stream, uint64 elem_count, double alpha, \
[all …]
|
| D | stream.cc | 2329 return impl(this, &blas::BlasSupport::DoBlasScal, elem_count, alpha, x, incx); in ThenBlasScal()
2337 return impl(this, &blas::BlasSupport::DoBlasScal, elem_count, alpha, x, incx); in ThenBlasScal()
2345 return impl(this, &blas::BlasSupport::DoBlasScal, elem_count, alpha, x, incx); in ThenBlasScal()
2353 return impl(this, &blas::BlasSupport::DoBlasScal, elem_count, alpha, x, incx); in ThenBlasScal()
2362 return impl(this, &blas::BlasSupport::DoBlasScal, elem_count, alpha, x, incx); in ThenBlasScal()
2371 return impl(this, &blas::BlasSupport::DoBlasScal, elem_count, alpha, x, incx); in ThenBlasScal()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 689 bool ROCMBlas::DoBlasScal(Stream *stream, uint64 elem_count, float alpha, in DoBlasScal() function in stream_executor::gpu::ROCMBlas
696 bool ROCMBlas::DoBlasScal(Stream *stream, uint64 elem_count, double alpha, in DoBlasScal() function in stream_executor::gpu::ROCMBlas
703 bool ROCMBlas::DoBlasScal(Stream *stream, uint64 elem_count, float alpha, in DoBlasScal() function in stream_executor::gpu::ROCMBlas
710 bool ROCMBlas::DoBlasScal(Stream *stream, uint64 elem_count, double alpha, in DoBlasScal() function in stream_executor::gpu::ROCMBlas
717 bool ROCMBlas::DoBlasScal(Stream *stream, uint64 elem_count, in DoBlasScal() function in stream_executor::gpu::ROCMBlas
725 bool ROCMBlas::DoBlasScal(Stream *stream, uint64 elem_count, in DoBlasScal() function in stream_executor::gpu::ROCMBlas
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 732 bool CUDABlas::DoBlasScal(Stream *stream, uint64 elem_count, float alpha, in DoBlasScal() function in stream_executor::gpu::CUDABlas
738 bool CUDABlas::DoBlasScal(Stream *stream, uint64 elem_count, double alpha, in DoBlasScal() function in stream_executor::gpu::CUDABlas
744 bool CUDABlas::DoBlasScal(Stream *stream, uint64 elem_count, float alpha, in DoBlasScal() function in stream_executor::gpu::CUDABlas
751 bool CUDABlas::DoBlasScal(Stream *stream, uint64 elem_count, double alpha, in DoBlasScal() function in stream_executor::gpu::CUDABlas
758 bool CUDABlas::DoBlasScal(Stream *stream, uint64 elem_count, in DoBlasScal() function in stream_executor::gpu::CUDABlas
766 bool CUDABlas::DoBlasScal(Stream *stream, uint64 elem_count, in DoBlasScal() function in stream_executor::gpu::CUDABlas
|