Searched refs:b_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
1826 b_raw_ptrs.push_back( in DoBlasGemmBatchedInternal()
1845 bsb = b_raw_ptrs[1] - b_raw_ptrs[0]; in DoBlasGemmBatchedInternal()
1859 long long iterative_bsb = b_raw_ptrs[i + 1] - b_raw_ptrs[i]; in DoBlasGemmBatchedInternal()
1892 GpuComplex(alpha_ptr), a_raw_ptrs[0], lda, bsa, b_raw_ptrs[0], ldb, bsb, 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
2154 b_raw_ptrs.push_back(static_cast<T *>(b_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
2203 !stream->ThenMemcpy(&b, b_raw_ptrs.data(), size).ok() || in DoBlasGemmBatchedInternal()
|