Searched refs:a_ptrs_to_wrappers (Results 1 – 2 of 2) sorted by relevance
2147 const port::ArraySlice<DeviceMemory<T> *> &a_ptrs_to_wrappers, int lda, in DoBlasGemmBatchedInternal() argument2153 a_raw_ptrs.push_back(static_cast<T *>(a_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()2258 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal()
1811 const port::ArraySlice<DeviceMemory<T> *> &a_ptrs_to_wrappers, int lda, in DoBlasGemmBatchedInternal() argument1825 reinterpret_cast<MAPPED_T *>(a_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()