Searched defs:layer_norm_scale_a (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 175 const int32_t* bias, int32_t layer_norm_scale_a, in ApplyLayerNorm()
185 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| D | portable_tensor_utils.cc | 412 const int32_t* bias, int32_t layer_norm_scale_a, in PortableApplyLayerNorm()
462 int32_t layer_norm_scale_a, in PortableApplyLayerNormFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.h | 151 const int32_t* bias, int32_t layer_norm_scale_a, in ApplyLayerNorm()
161 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| D | neon_tensor_utils.h | 143 const int32_t* bias, int32_t layer_norm_scale_a, in ApplyLayerNorm()
153 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| D | neon_tensor_utils.cc | 1492 const int32_t* bias, int32_t layer_norm_scale_a, in NeonApplyLayerNorm()
|