Home
last modified time | relevance | path

Searched refs:ComputationType (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dblas.cc73 string ComputationTypeString(ComputationType ty) { in ComputationTypeString()
75 case ComputationType::kF16: in ComputationTypeString()
77 case ComputationType::kF32: in ComputationTypeString()
79 case ComputationType::kF64: in ComputationTypeString()
81 case ComputationType::kI32: in ComputationTypeString()
83 case ComputationType::kComplexF32: in ComputationTypeString()
85 case ComputationType::kComplexF64: in ComputationTypeString()
92 std::ostream& operator<<(std::ostream& os, ComputationType ty) { in operator <<()
Dblas.h96 enum class ComputationType { enum
106 string ComputationTypeString(ComputationType ty);
108 std::ostream &operator<<(std::ostream &os, ComputationType ty);
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, \
[all …]
Dstream.h1329 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,
Dstream.cc82 string ToVlogString(blas::ComputationType ty) { in ToVlogString()
3792 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3804 int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3816 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3827 DeviceMemory<int> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3839 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3850 DeviceMemory<float> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3862 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3873 DeviceMemory<double> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3888 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc118 se::blas::ComputationType computation_type, in DoGemmWithAlgorithm()
160 se::blas::ComputationType computation_type, se::Stream* stream) { in DoGemmAutotune()
252 se::blas::ComputationType GetBlasComputationType(PrimitiveType type) { in GetBlasComputationType()
257 return se::blas::ComputationType::kF32; in GetBlasComputationType()
259 return se::blas::ComputationType::kF32; in GetBlasComputationType()
261 return se::blas::ComputationType::kF64; in GetBlasComputationType()
263 return se::blas::ComputationType::kComplexF32; in GetBlasComputationType()
265 return se::blas::ComputationType::kComplexF64; in GetBlasComputationType()
397 se::blas::ComputationType computation_type = in ExecuteOnStream()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.cc219 se::blas::ComputationType* compute_type) { in GetCublasAutotuneComputationType()
220 using se::blas::ComputationType; in GetCublasAutotuneComputationType()
226 *compute_type = ComputationType::kF32; in GetCublasAutotuneComputationType()
228 *compute_type = ComputationType::kF16; in GetCublasAutotuneComputationType()
232 *compute_type = ComputationType::kF32; in GetCublasAutotuneComputationType()
235 *compute_type = ComputationType::kF64; in GetCublasAutotuneComputationType()
258 using se::blas::ComputationType; in launch()
293 ComputationType computation_type; in launch()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc383 cudaDataType_t CUDAComputationType(blas::ComputationType ty) { in CUDAComputationType()
385 case blas::ComputationType::kF16: in CUDAComputationType()
387 case blas::ComputationType::kF32: in CUDAComputationType()
389 case blas::ComputationType::kF64: in CUDAComputationType()
391 case blas::ComputationType::kI32: in CUDAComputationType()
393 case blas::ComputationType::kComplexF32: in CUDAComputationType()
395 case blas::ComputationType::kComplexF64: in CUDAComputationType()
1892 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithmImpl()
2045 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2058 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
[all …]
Dcuda_blas.h125 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.h122 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type,
Drocm_blas.cc1702 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithmImpl()
1719 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1733 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
1746 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
1759 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
1774 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1789 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
/external/swiftshader/third_party/subzero/src/
DIceTargetLoweringMIPS32.h907 ComputationEntry(Inst *I, Type Ty) : Instr(I), ComputationType(Ty) {} in ComputationEntry()
917 Type ComputationType; variable
DIceTargetLoweringARM32.h1342 ComputationEntry(Inst *I, Type Ty) : Instr(I), ComputationType(Ty) {} in ComputationEntry()
1352 Type ComputationType; variable
DIceTargetLoweringMIPS32.cpp6061 switch (ComputationIter->second.ComputationType) { in recordProducers()
DIceTargetLoweringARM32.cpp7036 switch (ComputationIter->second.ComputationType) { in recordProducers()