Searched refs:b_ptrs_to_wrappers (Results 1 – 2 of 2) sorted by relevance
2148 const port::ArraySlice<DeviceMemory<T> *> &b_ptrs_to_wrappers, int ldb, in DoBlasGemmBatchedInternal() argument2154 b_raw_ptrs.push_back(static_cast<T *>(b_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()2259 const DeviceMemory<T> &b_matrix = *b_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal()
1812 const port::ArraySlice<DeviceMemory<T> *> &b_ptrs_to_wrappers, int ldb, in DoBlasGemmBatchedInternal() argument1827 reinterpret_cast<MAPPED_T *>(b_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()