Searched defs:computation_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1428 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
1444 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
1482 int64_t stride_c, int batch_count, blas::ComputationType computation_type, in ThenBlasGemmStridedBatchedWithAlgorithm()
2160 blas::ComputationType computation_type) { in CheckTypesForExtendedBlas()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 122 se::blas::ComputationType computation_type = in DoGemmWithAlgorithm() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 1280 optional<string> computation_type = in GetInstructionTrivialComputationStr() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2152 blas::DataType type_c, int ldc, blas::ComputationType computation_type, in DoBlasInternalImpl()
2182 int batch_count, blas::ComputationType computation_type, in DoBlasInternalImpl()
3131 blas::ComputationType computation_type, blas::DataType scale_type, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1593 blas::DataType type_c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
1605 int batch_count, blas::ComputationType computation_type, in DoBlasGemmStridedBatchedWithAlgorithm()
|