Searched refs:b_raw_ptrs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 2073 std::vector<MAPPED_T *> a_raw_ptrs, b_raw_ptrs, c_raw_ptrs; in DoBlasGemmBatchedInternal() local
2079 b_raw_ptrs.push_back( in DoBlasGemmBatchedInternal()
2099 b_raw_ptrs, batch_count, batch_stride_b, scratch_allocator, stream, in DoBlasGemmBatchedInternal()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2371 std::vector<T *> a_raw_ptrs, b_raw_ptrs, c_raw_ptrs; in DoBlasInternalImpl() local
2374 b_raw_ptrs.push_back(static_cast<T *>(b_ptrs_to_wrappers[i]->opaque())); in DoBlasInternalImpl()
2423 !stream->ThenMemcpy(&b, b_raw_ptrs.data(), size).ok() || in DoBlasInternalImpl()
|