Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc94 se::blas::ComputationType computation_type; in DoGemmWithAlgorithm() local
99 computation_type = se::blas::ComputationType::kF32; in DoGemmWithAlgorithm()
102 computation_type = se::blas::ComputationType::kF32; in DoGemmWithAlgorithm()
105 computation_type = se::blas::ComputationType::kF64; in DoGemmWithAlgorithm()
108 computation_type = se::blas::ComputationType::kComplexF32; in DoGemmWithAlgorithm()
111 computation_type = se::blas::ComputationType::kComplexF64; in DoGemmWithAlgorithm()
139 /*leading dim of output=*/output_matrix.num_rows, computation_type, in DoGemmWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2072 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl() argument
2173 CUDAComputationType(computation_type), in DoBlasInternalImpl()
2256 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl() argument
2260 computation_type, algorithm, output_profile_result); in DoBlasInternalImpl()
2269 int ldc, blas::ComputationType computation_type, in DoBlasInternalImpl() argument
2271 if (computation_type == blas::ComputationType::kF32) { in DoBlasInternalImpl()
2283 float_beta, c, ldc, computation_type, algorithm, output_profile_result); in DoBlasInternalImpl()
2286 CHECK_EQ(computation_type, blas::ComputationType::kF16); in DoBlasInternalImpl()
2289 computation_type, algorithm, output_profile_result); in DoBlasInternalImpl()
2297 int ldc, blas::ComputationType computation_type, in DoBlasInternalImpl() argument
[all …]
Dcuda_blas.h122 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.h173 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type,
Drocm_blas.cc1837 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithmImpl() argument
1854 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm() argument
1868 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
1881 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
1894 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm() argument
1909 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm() argument
1924 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm() argument
/external/tensorflow/tensorflow/stream_executor/
Dblas.h217 ComputationType computation_type; member
1109 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1117 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1124 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1131 int ldc, ComputationType computation_type, AlgorithmType algorithm,
1140 ComputationType computation_type, AlgorithmType algorithm,
1149 ComputationType computation_type, AlgorithmType algorithm,
2077 int ldc, blas::ComputationType computation_type, \
2088 blas::ComputationType computation_type, blas::AlgorithmType algorithm, \
2095 int ldc, blas::ComputationType computation_type, \
[all …]
Dstream.cc3435 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm() argument
3439 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(computation_type), in ThenBlasGemmWithAlgorithm()
3450 m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, computation_type, in ThenBlasGemmWithAlgorithm()
3459 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm() argument
3463 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(computation_type), in ThenBlasGemmWithAlgorithm()
3473 m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, computation_type, in ThenBlasGemmWithAlgorithm()
3482 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm() argument
3486 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(computation_type), in ThenBlasGemmWithAlgorithm()
3496 m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, computation_type, in ThenBlasGemmWithAlgorithm()
3505 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm() argument
[all …]
Dstream.h1352 int ldc, blas::ComputationType computation_type,
1360 int ldc, blas::ComputationType computation_type,
1368 int ldc, blas::ComputationType computation_type,
1376 int ldc, blas::ComputationType computation_type,
1386 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
1395 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc1275 optional<string> computation_type = in GetInstructionTrivialComputationStr() local
1277 if (!computation_type) { in GetInstructionTrivialComputationStr()
1282 HtmlLikeStringSanitize(*computation_type))); in GetInstructionTrivialComputationStr()
1285 HtmlLikeStringSanitize(*computation_type))); in GetInstructionTrivialComputationStr()