Searched defs:layer_norm_scale_a (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| 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()
|
| 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()
|
| D | neon_tensor_utils.cc | 1515 const int32_t* bias, int32_t layer_norm_scale_a, in NeonApplyLayerNorm()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| 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()
|
| 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()
|