Home
last modified time | relevance | path

Searched refs:c_ptrs_to_wrappers (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.h159 T beta, const port::ArraySlice<DeviceMemory<T> *> &c_ptrs_to_wrappers,
Drocm_blas.cc2044 T beta, const port::ArraySlice<DeviceMemory<T> *> &c_ptrs_to_wrappers, in DoBlasGemmBatchedInternal() argument
2082 reinterpret_cast<MAPPED_T *>(c_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2369 Scalar beta, const port::ArraySlice<DeviceMemory<T> *> &c_ptrs_to_wrappers, in DoBlasInternalImpl() argument
2375 c_raw_ptrs.push_back(static_cast<T *>(c_ptrs_to_wrappers[i]->opaque())); in DoBlasInternalImpl()
2501 DeviceMemory<T> *c_matrix = c_ptrs_to_wrappers[b]; in DoBlasInternalImpl()