Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dlayer_norm.cpp49 DimVector stat_shape; in layer_norm_with_mean_rstd_out() local
254 std::vector<int64_t> stat_shape; in math_native_layer_norm() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.cpp612 std::vector<int64_t> stat_shape; in compute_shape_native_layer_norm() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dlayer_norm_kernel.cu1373 std::vector<int64_t> stat_shape; in layer_norm_cuda() local