Home
last modified time | relevance | path

Searched refs:MAPPED_T (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1817 using MAPPED_T = typename EigenHalfToRocBlasHalf<T>::type; in DoBlasGemmBatchedInternal() typedef
1820 std::vector<MAPPED_T *> a_raw_ptrs, b_raw_ptrs, c_raw_ptrs; in DoBlasGemmBatchedInternal()
1825 reinterpret_cast<MAPPED_T *>(a_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
1827 reinterpret_cast<MAPPED_T *>(b_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
1829 reinterpret_cast<MAPPED_T *>(c_ptrs_to_wrappers[i]->opaque())); in DoBlasGemmBatchedInternal()
1885 MAPPED_T *alpha_ptr = reinterpret_cast<MAPPED_T *>(&alpha); in DoBlasGemmBatchedInternal()
1886 MAPPED_T *beta_ptr = reinterpret_cast<MAPPED_T *>(&beta); in DoBlasGemmBatchedInternal()