| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 118 se::blas::ComputationType computation_type, in DoGemmWithAlgorithm() argument
145 /*leading dim of output=*/output_matrix.num_rows, computation_type, in DoGemmWithAlgorithm()
160 se::blas::ComputationType computation_type, se::Stream* stream) { in DoGemmAutotune() argument
172 alpha, beta, computation_type, algorithm, in DoGemmAutotune()
397 se::blas::ComputationType computation_type = in ExecuteOnStream() local
436 beta_, computation_type, stream); in ExecuteOnStream()
458 computation_type, algorithm, stream, in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 293 ComputationType computation_type; in launch() local
295 GetCublasAutotuneComputationType(dtype, &computation_type); in launch()
313 &c_ptr, n, computation_type, profile_algorithm, in launch()
372 &c_ptr, n, computation_type, algorithm_config.algorithm(), in launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 1892 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithmImpl() argument
1962 CUDAComputationType(computation_type), in DoBlasGemmWithAlgorithmImpl()
2045 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm() argument
2049 computation_type, algorithm, output_profile_result); in DoBlasGemmWithAlgorithm()
2058 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
2060 if (computation_type == blas::ComputationType::kF32) { in DoBlasGemmWithAlgorithm()
2072 float_beta, c, ldc, computation_type, algorithm, output_profile_result); in DoBlasGemmWithAlgorithm()
2075 CHECK_EQ(computation_type, blas::ComputationType::kF16); in DoBlasGemmWithAlgorithm()
2078 computation_type, algorithm, output_profile_result); in DoBlasGemmWithAlgorithm()
2086 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
[all …]
|
| D | cuda_blas.h | 125 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 122 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type,
|
| D | rocm_blas.cc | 1702 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithmImpl() argument
1719 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm() argument
1733 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
1746 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
1759 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
1774 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm() argument
1789 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 1236 optional<string> computation_type = in GetInstructionTrivialComputationStr() local
1238 if (!computation_type) { in GetInstructionTrivialComputationStr()
1243 HtmlLikeStringSanitize(*computation_type))); in GetInstructionTrivialComputationStr()
1246 HtmlLikeStringSanitize(*computation_type))); in GetInstructionTrivialComputationStr()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 1040 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1048 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1055 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1062 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1071 ComputationType computation_type, AlgorithmType algorithm,
1080 ComputationType computation_type, AlgorithmType algorithm,
1941 int ldc, blas::ComputationType computation_type, \
1952 blas::ComputationType computation_type, blas::AlgorithmType algorithm, \
1959 int ldc, blas::ComputationType computation_type, \
1968 blas::ComputationType computation_type, blas::AlgorithmType algorithm, \
[all …]
|
| D | stream.cc | 3792 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm() argument
3796 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(computation_type), in ThenBlasGemmWithAlgorithm()
3807 m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, computation_type, in ThenBlasGemmWithAlgorithm()
3816 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm() argument
3820 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(computation_type), in ThenBlasGemmWithAlgorithm()
3830 m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, computation_type, in ThenBlasGemmWithAlgorithm()
3839 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm() argument
3843 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(computation_type), in ThenBlasGemmWithAlgorithm()
3853 m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, computation_type, in ThenBlasGemmWithAlgorithm()
3862 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm() argument
[all …]
|
| D | stream.h | 1329 int ldc, blas::ComputationType computation_type,
1337 int ldc, blas::ComputationType computation_type,
1345 int ldc, blas::ComputationType computation_type,
1353 int ldc, blas::ComputationType computation_type,
1363 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
1372 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|