Searched refs:a_ptrs_to_wrappers (Results 1 – 3 of 3) sorted by relevance
157 const port::ArraySlice<DeviceMemory<T> *> &a_ptrs_to_wrappers, int lda,
2042 const port::ArraySlice<DeviceMemory<T> *> &a_ptrs_to_wrappers, int lda, in DoBlasGemmBatchedInternal() argument2078 reinterpret_cast<MAPPED_T *>(a_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
2367 const port::ArraySlice<DeviceMemory<T> *> &a_ptrs_to_wrappers, int lda, in DoBlasInternalImpl() argument2373 a_raw_ptrs.push_back(static_cast<T *>(a_ptrs_to_wrappers[i]->opaque())); in DoBlasInternalImpl()2499 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasInternalImpl()