| /external/tensorflow/tensorflow/stream_executor/ |
| D | host_or_device_scalar.h | 29 class HostOrDeviceScalar {
33 HostOrDeviceScalar(ElemT value) : value_(value), is_pointer_(false) {} in HostOrDeviceScalar() function
34 explicit HostOrDeviceScalar(const DeviceMemory<ElemT>& pointer) in HostOrDeviceScalar() function
59 class HostOrDeviceScalar<void> {
67 HostOrDeviceScalar(float value) in HostOrDeviceScalar() function
70 HostOrDeviceScalar(double value) in HostOrDeviceScalar() function
73 HostOrDeviceScalar(Eigen::half value) in HostOrDeviceScalar() function
76 HostOrDeviceScalar(int8 value) in HostOrDeviceScalar() function
79 HostOrDeviceScalar(int32 value) in HostOrDeviceScalar() function
82 HostOrDeviceScalar(std::complex<float> value) in HostOrDeviceScalar() function
[all …]
|
| D | blas.h | 64 class HostOrDeviceScalar; variable
1106 uint64 n, uint64 k, const HostOrDeviceScalar<int> &alpha,
1108 int ldb, const HostOrDeviceScalar<int> &beta, DeviceMemory<int32> *c,
1113 uint64 n, uint64 k, const HostOrDeviceScalar<Eigen::half> &alpha,
1116 const HostOrDeviceScalar<Eigen::half> &beta, DeviceMemory<Eigen::half> *c,
1121 uint64 n, uint64 k, const HostOrDeviceScalar<float> &alpha,
1123 int ldb, const HostOrDeviceScalar<float> &beta, DeviceMemory<float> *c,
1128 uint64 n, uint64 k, const HostOrDeviceScalar<double> &alpha,
1130 int ldb, const HostOrDeviceScalar<double> &beta, DeviceMemory<double> *c,
1135 uint64 n, uint64 k, const HostOrDeviceScalar<std::complex<float>> &alpha,
[all …]
|
| D | stream.cc | 136 std::string ToVlogString(const HostOrDeviceScalar<T> &memory_or_constant) { in ToVlogString()
3431 uint64 k, const HostOrDeviceScalar<Eigen::half> &alpha, in ThenBlasGemmWithAlgorithm()
3434 const HostOrDeviceScalar<Eigen::half> &beta, DeviceMemory<Eigen::half> *c, in ThenBlasGemmWithAlgorithm()
3444 const HostOrDeviceScalar<Eigen::half> &, in ThenBlasGemmWithAlgorithm()
3446 int, const HostOrDeviceScalar<Eigen::half> &, DeviceMemory<Eigen::half> *, in ThenBlasGemmWithAlgorithm()
3456 uint64 k, const HostOrDeviceScalar<int> &alpha, const DeviceMemory<int8> &a, in ThenBlasGemmWithAlgorithm()
3458 const HostOrDeviceScalar<int> &beta, DeviceMemory<int> *c, int ldc, in ThenBlasGemmWithAlgorithm()
3468 const HostOrDeviceScalar<int> &, const DeviceMemory<int8> &, int, in ThenBlasGemmWithAlgorithm()
3469 const DeviceMemory<int8> &, int, const HostOrDeviceScalar<int> &, in ThenBlasGemmWithAlgorithm()
3479 uint64 k, const HostOrDeviceScalar<float> &alpha, in ThenBlasGemmWithAlgorithm()
[all …]
|
| D | stream.h | 1348 uint64 k, const HostOrDeviceScalar<Eigen::half> &alpha,
1351 const HostOrDeviceScalar<Eigen::half> &beta, DeviceMemory<Eigen::half> *c,
1357 uint64 k, const HostOrDeviceScalar<int> &alpha,
1359 int ldb, const HostOrDeviceScalar<int> &beta, DeviceMemory<int> *c,
1365 uint64 k, const HostOrDeviceScalar<float> &alpha,
1367 int ldb, const HostOrDeviceScalar<float> &beta, DeviceMemory<float> *c,
1373 uint64 k, const HostOrDeviceScalar<double> &alpha,
1375 int ldb, const HostOrDeviceScalar<double> &beta, DeviceMemory<double> *c,
1381 uint64 k, const HostOrDeviceScalar<std::complex<float>> &alpha,
1384 const HostOrDeviceScalar<std::complex<float>> &beta,
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 119 uint64 n, uint64 k, const HostOrDeviceScalar<CompT> &alpha,
121 const HostOrDeviceScalar<CompT> &beta, DeviceMemory<OutT> *c, int ldc,
145 const HostOrDeviceScalar<void> &alpha,
147 const HostOrDeviceScalar<void> &beta,
|
| D | cuda_blas.cc | 2069 uint64 n, uint64 k, const HostOrDeviceScalar<CompT> &alpha, in DoBlasInternalImpl()
2071 const HostOrDeviceScalar<CompT> &beta, DeviceMemory<OutT> *c, int ldc, in DoBlasInternalImpl()
2253 uint64 n, uint64 k, const HostOrDeviceScalar<int> &alpha, in DoBlasInternalImpl()
2255 const HostOrDeviceScalar<int> &beta, DeviceMemory<int> *c, int ldc, in DoBlasInternalImpl()
2265 uint64 n, uint64 k, const HostOrDeviceScalar<Eigen::half> &alpha, in DoBlasInternalImpl()
2268 const HostOrDeviceScalar<Eigen::half> &beta, DeviceMemory<Eigen::half> *c, in DoBlasInternalImpl()
2279 HostOrDeviceScalar<float> float_alpha(static_cast<float>(alpha.value())); in DoBlasInternalImpl()
2280 HostOrDeviceScalar<float> float_beta(static_cast<float>(beta.value())); in DoBlasInternalImpl()
2294 uint64 n, uint64 k, const HostOrDeviceScalar<float> &alpha, in DoBlasInternalImpl()
2296 int ldb, const HostOrDeviceScalar<float> &beta, DeviceMemory<float> *c, in DoBlasInternalImpl()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1851 uint64 n, uint64 k, const HostOrDeviceScalar<int> &alpha, in DoBlasGemmWithAlgorithm()
1853 const HostOrDeviceScalar<int> &beta, DeviceMemory<int32> *c, int ldc, in DoBlasGemmWithAlgorithm()
1864 uint64 n, uint64 k, const HostOrDeviceScalar<Eigen::half> &alpha, in DoBlasGemmWithAlgorithm()
1867 const HostOrDeviceScalar<Eigen::half> &beta, DeviceMemory<Eigen::half> *c, in DoBlasGemmWithAlgorithm()
1878 uint64 n, uint64 k, const HostOrDeviceScalar<float> &alpha, in DoBlasGemmWithAlgorithm()
1880 int ldb, const HostOrDeviceScalar<float> &beta, DeviceMemory<float> *c, in DoBlasGemmWithAlgorithm()
1891 uint64 n, uint64 k, const HostOrDeviceScalar<double> &alpha, in DoBlasGemmWithAlgorithm()
1893 int ldb, const HostOrDeviceScalar<double> &beta, DeviceMemory<double> *c, in DoBlasGemmWithAlgorithm()
1904 uint64 n, uint64 k, const HostOrDeviceScalar<std::complex<float>> &alpha, in DoBlasGemmWithAlgorithm()
1907 const HostOrDeviceScalar<std::complex<float>> &beta, in DoBlasGemmWithAlgorithm()
[all …]
|