Home
last modified time | relevance | path

Searched defs:X_vec (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
Dlayer_norm_kernel.cu164 const vec_t * X_vec = reinterpret_cast<const vec_t*>(X); in compute_stats() local
237 const vec_t * X_vec = reinterpret_cast<const vec_t*>(block_row); in vectorized_layer_norm_kernel_impl() local