Searched defs:layer_norm_weights (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| 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()
|
| 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()
|
| D | neon_tensor_utils.cc | 1514 void NeonApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in NeonApplyLayerNorm()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| 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()
|
| D | portable_tensor_utils.cc | 446 const int16_t* layer_norm_weights, in PortableApplyLayerNorm()
496 const int16_t* layer_norm_weights, in PortableApplyLayerNormFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| 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
|