Searched defs:layer_norm_weights (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | neon_tensor_utils.h | 157 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in ApplyLayerNorm()
167 const int16_t* layer_norm_weights, in ApplyLayerNormFloat()
|
| H A D | sse_tensor_utils.h | 170 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in ApplyLayerNorm()
180 const int16_t* layer_norm_weights, in ApplyLayerNormFloat()
|
| H A D | neon_tensor_utils.cc | 1514 void NeonApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in NeonApplyLayerNorm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | portable_tensor_utils.h | 189 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in ApplyLayerNorm()
199 const int16_t* layer_norm_weights, in ApplyLayerNormFloat()
|
| H A D | portable_tensor_utils.cc | 446 const int16_t* layer_norm_weights, in PortableApplyLayerNorm()
496 const int16_t* layer_norm_weights, in PortableApplyLayerNormFloat()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.cpp | 10 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, const int32_t* bias, in ApplyLayerNorm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | tensor_utils_test.cc | 705 const std::vector<int16_t> layer_norm_weights = { in TEST() local
737 const std::vector<int16_t> layer_norm_weights = { in TEST() local
|