Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dblas.cc75 std::string ComputationTypeString(ComputationType ty) { in ComputationTypeString()
77 case ComputationType::kF16: in ComputationTypeString()
79 case ComputationType::kF32: in ComputationTypeString()
81 case ComputationType::kF64: in ComputationTypeString()
83 case ComputationType::kI32: in ComputationTypeString()
85 case ComputationType::kComplexF32: in ComputationTypeString()
87 case ComputationType::kComplexF64: in ComputationTypeString()
94 std::ostream& operator<<(std::ostream& os, ComputationType ty) { in operator <<()
Dblas.h100 enum class ComputationType { enum
122 std::string ComputationTypeString(ComputationType ty);
124 std::ostream &operator<<(std::ostream &os, ComputationType ty);
217 ComputationType computation_type;
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,
[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,
Dstream.cc83 std::string ToVlogString(blas::ComputationType ty) { in ToVlogString()
3435 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3447 int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3459 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3470 DeviceMemory<int> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3482 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3493 DeviceMemory<float> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3505 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3516 DeviceMemory<double> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3531 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc94 se::blas::ComputationType computation_type; in DoGemmWithAlgorithm()
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()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc399 cudaDataType_t CUDAComputationType(blas::ComputationType ty) { in CUDAComputationType()
401 case blas::ComputationType::kF16: in CUDAComputationType()
403 case blas::ComputationType::kF32: in CUDAComputationType()
405 case blas::ComputationType::kF64: in CUDAComputationType()
407 case blas::ComputationType::kI32: in CUDAComputationType()
409 case blas::ComputationType::kComplexF32: in CUDAComputationType()
411 case blas::ComputationType::kComplexF64: in CUDAComputationType()
413 case blas::ComputationType::kTF32AsF32: // fall-through in CUDAComputationType()
414 case blas::ComputationType::kBF16AsF32: in CUDAComputationType()
422 cublasComputeType_t CUBLASComputationType(blas::ComputationType ty) { in CUBLASComputationType()
[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()
1854 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1868 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
1881 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
1894 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
1909 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1924 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
/external/swiftshader/third_party/subzero/src/
DIceTargetLoweringMIPS32.h901 ComputationEntry(Inst *I, Type Ty) : Instr(I), ComputationType(Ty) {} in ComputationEntry()
911 Type ComputationType; variable
DIceTargetLoweringARM32.h1335 ComputationEntry(Inst *I, Type Ty) : Instr(I), ComputationType(Ty) {} in ComputationEntry()
1345 Type ComputationType; variable
DIceTargetLoweringMIPS32.cpp6025 switch (ComputationIter->second.ComputationType) { in recordProducers()
DIceTargetLoweringARM32.cpp7016 switch (ComputationIter->second.ComputationType) { in recordProducers()