Searched refs:c_matrix (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gemm_functors.h | 103 typename tensorflow::TTypes<T3>::Matrix c_matrix(c, m, n); in operator()
108 c_matrix.device(ctx->eigen_device<Eigen::ThreadPoolDevice>()) = in operator()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2260 DeviceMemory<T> *c_matrix = c_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal() local
2262 lda, b_matrix, ldb, beta, c_matrix, ldc); in DoBlasGemmBatchedInternal()
2399 auto *c_matrix = in DoBlasGemmStridedBatched() local
2405 lda, b_matrix, SE_CUDA_DATA_HALF, ldb, &beta, c_matrix, in DoBlasGemmStridedBatched()
|