/external/tensorflow/tensorflow/stream_executor/ |
D | blas.cc | 75 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 <<()
|
D | blas.h | 100 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 …]
|
D | stream.h | 1352 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,
|
D | stream.cc | 83 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/ |
D | gemm_thunk.cc | 94 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/ |
D | cuda_blas.cc | 399 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 …]
|
D | cuda_blas.h | 122 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_blas.h | 173 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type,
|
D | rocm_blas.cc | 1837 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/ |
D | IceTargetLoweringMIPS32.h | 901 ComputationEntry(Inst *I, Type Ty) : Instr(I), ComputationType(Ty) {} in ComputationEntry() 911 Type ComputationType; variable
|
D | IceTargetLoweringARM32.h | 1335 ComputationEntry(Inst *I, Type Ty) : Instr(I), ComputationType(Ty) {} in ComputationEntry() 1345 Type ComputationType; variable
|
D | IceTargetLoweringMIPS32.cpp | 6025 switch (ComputationIter->second.ComputationType) { in recordProducers()
|
D | IceTargetLoweringARM32.cpp | 7016 switch (ComputationIter->second.ComputationType) { in recordProducers()
|