Searched refs:c_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
2081 c_raw_ptrs.push_back( in DoBlasGemmBatchedInternal()
2108 c_raw_ptrs, batch_count, batch_stride_c, scratch_allocator, stream, in DoBlasGemmBatchedInternal()
2128 return ReorganizeMemory(stream, &c, c_raw_ptrs, batch_count, batch_stride_c, 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
2375 c_raw_ptrs.push_back(static_cast<T *>(c_ptrs_to_wrappers[i]->opaque())); in DoBlasInternalImpl()
2424 !stream->ThenMemcpy(&c, c_raw_ptrs.data(), size).ok()) { in DoBlasInternalImpl()
|