Home
last modified time | relevance | path

Searched defs:layer_norm_scale (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils_test.cc52 SmallVector<int64_t, 1> layer_norm_scale{4}; in createLstmCompositeFunc() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dportable_tensor_utils.cc503 const float layer_norm_scale = in PortableApplyLayerNormFloat() local