| /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()
|