Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for IsStrCompositeAttribute (0.26 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td

                        [(IsBoolCompositeAttribute<"ceil_mode", "false"> $attrs),
                        (IsBoolCompositeAttribute<"count_include_pad", "false"> $attrs),
                        (IsStrCompositeAttribute<"divisor_override", "py_None"> $attrs),
                        (HasSameOrValidPadding $old_val)]>;
    
    // Replaces an aten.avg_pool2d with (T -> tfl.pad -> tfl.average_pool_2d  -> T).
    def LegalizeAvgPool2dWithPadComposite: Pat<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 23:16:05 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_lowering_patterns.td

            (TFL_GeluOp $inputs, ConstBoolAttrFalse),
            [(IsStrCompositeAttribute<"approximate", "none"> $attrs)]>;
    
    def LegalizeCompositeApproximateAtenGELU : Pat<
            (MHLO_CompositeOp:$composite
              (variadic $inputs),
              ConstantStrAttr<StrAttr, "aten.gelu.default">, $attrs, $_, $_),
            (TFL_GeluOp $inputs, ConstBoolAttrTrue),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 18:45:51 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.td

      IsCompositeAttribute<attr_name, BoolAttr, val>;
    
    // Receives a composite DictionaryAttr as an argument and checks if has a
    // StrAttr with the name `attr_name` and value `val`.
    class IsStrCompositeAttribute <string attr_name, string val>:
      IsCompositeAttribute<attr_name, StrAttr, "\"" # val # "\"">;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 19:05:30 UTC 2024
    - 2.7K bytes
    - Viewed (0)
Back to top