Searched defs:layer_norm_weights (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 174 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in ApplyLayerNorm()
184 const int16_t* layer_norm_weights, in ApplyLayerNormFloat()
|
| D | portable_tensor_utils.cc | 411 const int16_t* layer_norm_weights, in PortableApplyLayerNorm()
461 const int16_t* layer_norm_weights, in PortableApplyLayerNormFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | neon_tensor_utils.h | 142 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in ApplyLayerNorm()
152 const int16_t* layer_norm_weights, in ApplyLayerNormFloat()
|
| D | sse_tensor_utils.h | 150 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in ApplyLayerNorm()
160 const int16_t* layer_norm_weights, in ApplyLayerNormFloat()
|
| D | neon_tensor_utils.cc | 1491 void NeonApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, in NeonApplyLayerNorm()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_test.cc | 703 const std::vector<int16_t> layer_norm_weights = { in TEST() local
735 const std::vector<int16_t> layer_norm_weights = { in TEST() local
|