Searched defs:layer_norm_scale_b (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 176 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm()
185 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| D | portable_tensor_utils.cc | 413 int32_t layer_norm_scale_b, int32_t variance_limit, in PortableApplyLayerNorm()
463 int32_t layer_norm_scale_b, in PortableApplyLayerNormFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.h | 152 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm()
161 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| D | neon_tensor_utils.h | 144 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm()
153 int32_t layer_norm_scale_a, int32_t layer_norm_scale_b, in ApplyLayerNormFloat()
|
| D | neon_tensor_utils.cc | 1493 int32_t layer_norm_scale_b, int32_t variance_limit, in NeonApplyLayerNorm()
|