Searched refs:smem_input_cache_vec_ptr (Results 1 – 1 of 1) sorted by relevance
718 LoadT* const smem_input_cache_vec_ptr = reinterpret_cast<LoadT*>(smem_input_cache); in cunn_SoftMaxForwardSmem() local725 smem_input_cache_vec_ptr[offset] = crnt_vec; in cunn_SoftMaxForwardSmem()740 LoadT crnt_vec = smem_input_cache_vec_ptr[offset]; in cunn_SoftMaxForwardSmem()757 LoadT crnt_vec = smem_input_cache_vec_ptr[offset]; in cunn_SoftMaxForwardSmem()