Searched refs:b_ptrs_to_wrappers (Results 1 – 3 of 3) sorted by relevance
158 const port::ArraySlice<DeviceMemory<T> *> &b_ptrs_to_wrappers, int ldb,
2043 const port::ArraySlice<DeviceMemory<T> *> &b_ptrs_to_wrappers, int ldb, in DoBlasGemmBatchedInternal() argument2080 reinterpret_cast<MAPPED_T *>(b_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
2368 const port::ArraySlice<DeviceMemory<T> *> &b_ptrs_to_wrappers, int ldb, in DoBlasInternalImpl() argument2374 b_raw_ptrs.push_back(static_cast<T *>(b_ptrs_to_wrappers[i]->opaque())); in DoBlasInternalImpl()2500 const DeviceMemory<T> &b_matrix = *b_ptrs_to_wrappers[b]; in DoBlasInternalImpl()