Searched defs:computation_type (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 118 se::blas::ComputationType computation_type, in DoGemmWithAlgorithm()
160 se::blas::ComputationType computation_type, se::Stream* stream) { in DoGemmAutotune()
397 se::blas::ComputationType computation_type = in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 293 ComputationType computation_type; in launch() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1702 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/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 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()
2086 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
2098 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
2112 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2126 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 1236 optional<string> computation_type = in GetInstructionTrivialComputationStr() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 3792 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3816 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3839 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3862 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3888 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3916 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
|