Searched refs:a_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
2077 a_raw_ptrs.push_back( in DoBlasGemmBatchedInternal()
2090 a_raw_ptrs, batch_count, batch_stride_a, 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
2373 a_raw_ptrs.push_back(static_cast<T *>(a_ptrs_to_wrappers[i]->opaque())); in DoBlasInternalImpl()
2422 if (!stream->ThenMemcpy(&a, a_raw_ptrs.data(), size).ok() || in DoBlasInternalImpl()
|