Searched defs:computation_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 93 se::blas::ComputationType computation_type, in DoGemmWithAlgorithm()
128 MatrixDescriptor output_matrix, se::blas::ComputationType computation_type, in DoGemmAutotune()
285 se::blas::ComputationType computation_type = in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 298 ComputationType computation_type; in launch() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2155 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithmImpl()
2244 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
2257 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2268 int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
2279 DeviceMemory<double> *c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
2292 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2305 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 1165 optional<string> computation_type = in GetInstructionTrivialComputationStr() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 3789 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3812 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3833 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3854 DeviceMemory<double> *c, int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3878 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3903 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
|