Searched defs:LoadT (Results 1 – 3 of 3) sorted by relevance
460 using LoadT = at::native::memory::aligned_vector<T, ILP>; in ilpReduce() typedef507 using LoadT = at::native::memory::aligned_vector<scalar_t, ILP>; in WriteFpropResultsVectorized() typedef716 using LoadT = at::native::memory::aligned_vector<scalar_t, ILP>; in cunn_SoftMaxForwardSmem() typedef777 using LoadT = at::native::memory::aligned_vector<scalar_t, ILP>; in cunn_SoftMaxBackward() typedef
56 using LoadT = memory::aligned_vector<scalar_t, VEC>; in fused_dropout_kernel_vec() typedef
119 using LoadT = memory::aligned_vector<scalar_t, VEC>; in transform_bias_rescale_qkv_kernel() typedef226 using LoadT = memory::aligned_vector<scalar_t, VEC>; in transform_bias_rescale_qkv_add_padding_kernel() typedef