Searched refs:variance_limit (Results 1 – 8 of 8) 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() argument
179 layer_norm_scale_b, variance_limit, n_batch, n_input, in ApplyLayerNorm()
|
| D | portable_tensor_utils_impl.h | 142 int32_t layer_norm_scale_b, int32_t variance_limit,
|
| D | portable_tensor_utils.cc | 413 int32_t layer_norm_scale_b, int32_t variance_limit, in PortableApplyLayerNorm() argument
435 variance2 = variance_limit; in PortableApplyLayerNorm()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | neon_tensor_utils.h | 144 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm() argument
147 layer_norm_scale_a, layer_norm_scale_b, variance_limit, in ApplyLayerNorm()
|
| D | sse_tensor_utils.h | 152 int32_t layer_norm_scale_b, int32_t variance_limit, in ApplyLayerNorm() argument
155 layer_norm_scale_b, variance_limit, n_batch, n_input, in ApplyLayerNorm()
|
| D | neon_tensor_utils_impl.h | 64 int32_t layer_norm_scale_b, int32_t variance_limit,
|
| D | neon_tensor_utils.cc | 1493 int32_t layer_norm_scale_b, int32_t variance_limit, in NeonApplyLayerNorm() argument
1532 variance2 = variance_limit; in NeonApplyLayerNorm()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_common.h | 205 int32_t layer_norm_scale_b, int32_t variance_limit,
|