Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for indy_lstm_ (0.32 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc

      // function attribute prefixed with `"indy_lstm_"`.
      // IndyLSTMs have diagonal recurrent weight matrices and can benefit from
      // more efficent operations in TFLite with the correct conversion (i.e. when
      // the diagonal recurrent weight matrices are provided as vectors).
      if (attr.getValue().starts_with("indy_lstm_")) {
        // Check if the keras lstm can be fused, if not, we just don't do anything.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

                                         &recurrent_weights_array)))
        return failure();
    
      // Reshape recurrent weights to vectors if indy behaviour is enabled.
      // IndyLSTMs are a LSTM variant with diagonal recurrent weight
      // matrices. For optimization purposes these are provided as vectors.
      Value recurrent_to_input_weights =
          indy ? mlir::cast<Value>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
Back to top