Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for LSTMKernelTypeAttr (0.19 sec)

  1. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

          output_layer_norm_coefficients_, builder_.getStringAttr("TANH"),
          builder_.getF32FloatAttr(10.0), builder_.getF32FloatAttr(0.0),
          mlir::TFL::LSTMKernelTypeAttr::get(builder_.getContext(),
                                             mlir::TFL::LSTMKernelType::FULL),
          /*asymmetric_quantize_inputs=*/mlir::BoolAttr(),
          /*input_to_input_intermediate=*/mlir::TypeAttr(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

          kernel_type = mlir::TFL::LSTMKernelType::FULL;
          break;
        case tflite::LSTMKernelType_BASIC:
          kernel_type = mlir::TFL::LSTMKernelType::BASIC;
          break;
      }
      return mlir::TFL::LSTMKernelTypeAttr::get(builder.getContext(), kernel_type);
    }
    
    static mlir::Attribute BuildTFL_MirrorPaddingAttr(tflite::MirrorPadMode value,
                                                      mlir::Builder builder) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        "whose value is " # n>;
    
    // Ensures the given LSTMKernelType attribute has the given value.
    class TFL_LSTMKernelTypeEqualsTo<string value> : AttrConstraint<
        CPred<"$_self.isa<LSTMKernelTypeAttr>() && "
                "$_self.cast<LSTMKernelTypeAttr>().getValue() == " # value>,
        "whose value is " # value>;
    
    // This is a quantization-aware version of TCresVTEtIsSameAsOp
    class TFL_TCresVTEtIsSameAsOp<int i, int j> : And<[
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top