Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for lstm_ (0.09 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let summary = "Bidirectional sequence lstm operator";
    
      let description = [{
        Bidirectional lstm is essentially two lstms, one running forward & the
        other running backward. And the output is the concatenation of the two
        lstms.
      }];
    
      let arguments = (
        ins TFL_TensorOf<[F32, I8]>:$input,
    
        // Forward LSTM Weights
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/ops.mlir

      // expected-error @+1 {{'tfl.lstm' op failed to verify that either projection weight must be specified or both projection weight and projection bias must not be specified}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 189.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir

    // CHECK-DAG:       [[VAL_47:%.*]] = arith.constant dense<0.000000e+00> : tensor<1x3xf32>
    // CHECK-DAG:       [[VAL_48:%.*]] = arith.constant dense<0.000000e+00> : tensor<1x1xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 122.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      // duplicated buffer is found.
      std::optional<BufferOffset<tflite::Buffer>> BuildBuffer(
          Value value, bool can_be_deduplicated, int& index);
    
      // Build TFLite tensor from the given type. This function is for tfl.lstm
      // intermediates, which should have UniformQuantizedType.
      std::optional<BufferOffset<tflite::Tensor>> BuildTensorFromType(
          mlir::Type type, const std::string& name);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  5. 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)
Back to top