Searched defs:computation_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.h | 869 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
885 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
923 int64_t stride_c, int batch_count, blas::ComputationType computation_type, in ThenBlasGemmStridedBatchedWithAlgorithm()
1490 blas::ComputationType computation_type) { in CheckTypesForExtendedBlas()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_blas_lt.cc | 132 blas::DataType c_type, blas::ComputationType computation_type) { in GetScaleType()
|
| D | cuda_blas.cc | 1020 blas::DataType type_c, int ldc, blas::ComputationType computation_type, in DoBlasInternalImpl()
1050 int batch_count, blas::ComputationType computation_type, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 634 blas::DataType type_c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithm()
646 int batch_count, blas::ComputationType computation_type, in DoBlasGemmStridedBatchedWithAlgorithm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 1437 optional<std::string> computation_type = in GetInstructionTrivialComputationStr() local
|