Home
last modified time | relevance | path

Searched defs:computation_type (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc94 se::blas::ComputationType computation_type; in DoGemmWithAlgorithm() local
/external/tensorflow/tensorflow/stream_executor/
Dblas.h217 ComputationType computation_type; member
Dstream.cc3435 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3459 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3482 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3505 int ldc, blas::ComputationType computation_type, in ThenBlasGemmWithAlgorithm()
3531 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3559 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2072 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
2256 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
2269 int ldc, blas::ComputationType computation_type, in DoBlasInternalImpl()
2297 int ldc, blas::ComputationType computation_type, in DoBlasInternalImpl()
2309 int ldc, blas::ComputationType computation_type, in DoBlasInternalImpl()
2323 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
2337 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
3179 blas::ComputationType computation_type, blas::DataType scale_type, in DoBlasInternalImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1837 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/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc1275 optional<string> computation_type = in GetInstructionTrivialComputationStr() local