- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for forget_layer_norm_coefficients (0.2 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
if (is_layer_norm_lstm) { const auto forget_layer_norm_coefficients = op.getForgetLayerNormCoefficients().getType().cast<ShapedType>(); // If this lstm has layer normalization, this input value, // "forget_layer_norm_coefficients" should be a 1D tensor. if (!forget_layer_norm_coefficients.hasRank() || forget_layer_norm_coefficients.getRank() != 1 ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
TFL_StatefulTensor:$input_cell_state, // Layer norm coefficients TFL_TensorOfOrNone<[F32, QI16]>:$input_layer_norm_coefficients, TFL_TensorOfOrNone<[F32, QI16]>:$forget_layer_norm_coefficients, TFL_TensorOfOrNone<[F32, QI16]>:$cell_layer_norm_coefficients, TFL_TensorOfOrNone<[F32, QI16]>:$output_layer_norm_coefficients, // Attributes TFL_AFAttr:$fused_activation_function,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)