Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for HasValidPrecisionConfig (0.37 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td

    def ToPrecisionConfigsAttr : NativeCodeCall<"GetPrecisionConfigAttr($0, &$_builder)">;
    
    def HasValidDotDims : Constraint<CPred<"HasValidDotDims($0)">>;
    
    def HasValidPrecisionConfig : Constraint<CPred<"HasValidPrecisionConfig($0)">>;
    
    def : Pat<(TF_XlaDotOp $lhs, $rhs, $dimension_numbers, $precision_config),
              (MHLO_DotGeneralOp $lhs, $rhs,
                                (ToDotDimNumsAttr $dimension_numbers),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 34.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

      ::xla::DotDimensionNumbers dims;
      if (!dims.ParseFromString(attr.getValue().str())) return {};
      return ::xla::ConvertDotDimensionNumbers(dims, builder);
    }
    
    bool HasValidPrecisionConfig(StringAttr attr) {
      ::xla::PrecisionConfig precision;
      return precision.ParseFromString(attr.getValue().str());
    }
    
    mlir::ArrayAttr GetPrecisionConfigAttr(StringAttr attr, Builder *builder) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top