Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dnormalization_ops_util.cpp120 Tensor::SizesType* mean_rstd_sizes, in get_layer_norm_out_target_size()
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_native_layer_norm.cpp147 Tensor::SizesType mean_rstd_sizes[kTensorDimensionLimit]; in native_layer_norm_out() local
H A Dop_native_group_norm.cpp139 Tensor::SizesType mean_rstd_sizes[kTensorDimensionLimit]; in native_group_norm_out() local
/aosp_15_r20/external/executorch/kernels/optimized/cpu/
H A Dop_native_layer_norm.cpp135 Tensor::SizesType mean_rstd_sizes[kTensorDimensionLimit]; in opt_native_layer_norm_out() local