Searched refs:c_raw_ptrs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1820 std::vector<MAPPED_T *> a_raw_ptrs, b_raw_ptrs, c_raw_ptrs; in DoBlasGemmBatchedInternal() local
1828 c_raw_ptrs.push_back( in DoBlasGemmBatchedInternal()
1846 bsc = c_raw_ptrs[1] - c_raw_ptrs[0]; in DoBlasGemmBatchedInternal()
1865 long long iterative_bsc = c_raw_ptrs[i + 1] - c_raw_ptrs[i]; in DoBlasGemmBatchedInternal()
1893 GpuComplex(beta_ptr), c_raw_ptrs[0], ldc, bsc, batch_count); in DoBlasGemmBatchedInternal()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2151 std::vector<T *> a_raw_ptrs, b_raw_ptrs, c_raw_ptrs; in DoBlasGemmBatchedInternal() local
2155 c_raw_ptrs.push_back(static_cast<T *>(c_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
2204 !stream->ThenMemcpy(&c, c_raw_ptrs.data(), size).ok()) { in DoBlasGemmBatchedInternal()
|