Searched refs:c_ptrs_to_wrappers (Results 1 – 2 of 2) sorted by relevance
2149 Scalar beta, const port::ArraySlice<DeviceMemory<T> *> &c_ptrs_to_wrappers, in DoBlasGemmBatchedInternal() argument2155 c_raw_ptrs.push_back(static_cast<T *>(c_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()2260 DeviceMemory<T> *c_matrix = c_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal()
1813 T beta, const port::ArraySlice<DeviceMemory<T> *> &c_ptrs_to_wrappers, in DoBlasGemmBatchedInternal() argument1829 reinterpret_cast<MAPPED_T *>(c_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()