| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 384 virtual bool DoBlasIamax(Stream *stream, uint64 elem_count,
387 virtual bool DoBlasIamax(Stream *stream, uint64 elem_count,
390 virtual bool DoBlasIamax(Stream *stream, uint64 elem_count,
393 virtual bool DoBlasIamax(Stream *stream, uint64 elem_count,
1535 bool DoBlasIamax(Stream *stream, uint64 elem_count, \
1538 bool DoBlasIamax(Stream *stream, uint64 elem_count, \
1541 bool DoBlasIamax(Stream *stream, uint64 elem_count, \
1544 bool DoBlasIamax(Stream *stream, uint64 elem_count, \
|
| D | stream.cc | 2422 return impl(this, &blas::BlasSupport::DoBlasIamax, elem_count, x, incx, in ThenBlasIamax()
2432 return impl(this, &blas::BlasSupport::DoBlasIamax, elem_count, x, incx, in ThenBlasIamax()
2443 return impl(this, &blas::BlasSupport::DoBlasIamax, elem_count, x, incx, in ThenBlasIamax()
2454 return impl(this, &blas::BlasSupport::DoBlasIamax, elem_count, x, incx, in ThenBlasIamax()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 765 bool ROCMBlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::ROCMBlas
773 bool ROCMBlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::ROCMBlas
781 bool ROCMBlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::ROCMBlas
789 bool ROCMBlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::ROCMBlas
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 806 bool CUDABlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::CUDABlas
814 bool CUDABlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::CUDABlas
822 bool CUDABlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::CUDABlas
830 bool CUDABlas::DoBlasIamax(Stream *stream, uint64 elem_count, in DoBlasIamax() function in stream_executor::gpu::CUDABlas
|