Searched defs:layer_norm_scale_a (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | neon_tensor_utils.h | 158 const int32_t* bias, int32_t layer_norm_scale_a, in ApplyLayerNorm()
168 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| H A D | sse_tensor_utils.h | 171 const int32_t* bias, int32_t layer_norm_scale_a, in ApplyLayerNorm()
181 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| H A D | neon_tensor_utils.cc | 1515 const int32_t* bias, int32_t layer_norm_scale_a, in NeonApplyLayerNorm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | portable_tensor_utils.h | 190 const int32_t* bias, int32_t layer_norm_scale_a, in ApplyLayerNorm()
200 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| H A D | portable_tensor_utils.cc | 447 const int32_t* bias, int32_t layer_norm_scale_a, in PortableApplyLayerNorm()
497 int32_t layer_norm_scale_a, in PortableApplyLayerNormFloat()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.cpp | 11 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm()
|