Searched defs:layer_norm_scale_b (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 | 159 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm()
168 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| H A D | sse_tensor_utils.h | 172 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm()
181 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| H A D | neon_tensor_utils.cc | 1516 int32_t layer_norm_scale_b, int32_t variance_limit, in NeonApplyLayerNorm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | portable_tensor_utils.h | 191 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm()
200 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| H A D | portable_tensor_utils.cc | 448 int32_t layer_norm_scale_b, int32_t variance_limit, in PortableApplyLayerNorm()
498 int32_t layer_norm_scale_b, 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()
|