Searched refs:kNoTranspose (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | triangular_solve_thunk.cc | 58 return se::blas::Transpose::kNoTranspose; in TriangularSolveThunk()
66 return se::blas::Transpose::kNoTranspose; in TriangularSolveThunk()
|
| D | gemm_thunk.cc | 122 : se::blas::Transpose::kNoTranspose; in DoGemmWithAlgorithm()
124 : se::blas::Transpose::kNoTranspose; in DoGemmWithAlgorithm()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | blas_gemm.cc | 45 se::blas::Transpose trans[] = {se::blas::Transpose::kNoTranspose, in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.cc | 25 case Transpose::kNoTranspose: in TransposeString()
|
| D | blas.h | 70 enum class Transpose { kNoTranspose, kTranspose, kConjugateTranspose }; enumerator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 327 se::blas::Transpose trans[] = {se::blas::Transpose::kNoTranspose,
421 ? se::blas::Transpose::kNoTranspose
499 se::blas::Transpose trans[] = {se::blas::Transpose::kNoTranspose,
|
| D | conv_grad_input_ops.cc | 149 auto no_transpose = se::blas::Transpose::kNoTranspose; in operator ()()
182 auto no_transpose = se::blas::Transpose::kNoTranspose; in operator ()()
|
| D | conv_ops_3d.cc | 267 auto no_transpose = se::blas::Transpose::kNoTranspose; in launch()
294 auto no_transpose = se::blas::Transpose::kNoTranspose; in launch()
|
| D | conv_grad_ops_3d.cc | 1199 auto no_transpose = se::blas::Transpose::kNoTranspose; in Compute()
1229 auto no_transpose = se::blas::Transpose::kNoTranspose; in Compute()
1694 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in Compute()
1722 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in Compute()
|
| D | conv_grad_filter_ops.cc | 744 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in operator ()()
775 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in operator ()()
|
| D | conv_ops.cc | 694 auto no_transpose = se::blas::Transpose::kNoTranspose; in operator ()()
722 auto no_transpose = se::blas::Transpose::kNoTranspose; in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 291 case blas::Transpose::kNoTranspose: in CUDABlasTranspose()
1751 if (transa == blas::Transpose::kNoTranspose) { in DoBlasInternalImpl()
1762 if (transb == blas::Transpose::kNoTranspose) { in DoBlasInternalImpl()
1805 if (transa == blas::Transpose::kNoTranspose) { in DoBlasInternalImpl()
1816 if (transb == blas::Transpose::kNoTranspose) { in DoBlasInternalImpl()
3268 uint64 rows_a = p.transa == blas::Transpose::kNoTranspose ? p.m : p.k; in DoBlasInternalImpl()
3269 uint64 cols_a = p.transa == blas::Transpose::kNoTranspose ? p.k : p.m; in DoBlasInternalImpl()
3270 uint64 rows_b = p.transb == blas::Transpose::kNoTranspose ? p.k : p.n; in DoBlasInternalImpl()
3271 uint64 cols_b = p.transb == blas::Transpose::kNoTranspose ? p.n : p.k; in DoBlasInternalImpl()
|
| D | cuda_dnn.cc | 4105 stream->ThenBlasGemm(blas::Transpose::kNoTranspose, in DoMatMul()
4106 blas::Transpose::kNoTranspose, m, n, k, alpha, weights, in DoMatMul()
4184 stream->ThenBlasGemmBatched(blas::Transpose::kNoTranspose, in DoMatMul()
4185 blas::Transpose::kNoTranspose, m, n, k, alpha, in DoMatMul()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 379 case blas::Transpose::kNoTranspose: in ROCMBlasTranspose()
1598 if (transa == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1609 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1644 if (transa == blas::Transpose::kNoTranspose) { in DoBlasGemm()
1655 if (transb == blas::Transpose::kNoTranspose) { in DoBlasGemm()
|
| D | rocm_dnn.cc | 3858 stream->ThenBlasGemm(blas::Transpose::kNoTranspose, in DoMatMul()
3859 blas::Transpose::kNoTranspose, m, n, k, alpha, weights, in DoMatMul()
3937 stream->ThenBlasGemmBatched(blas::Transpose::kNoTranspose, in DoMatMul()
3938 blas::Transpose::kNoTranspose, m, n, k, alpha, in DoMatMul()
|