Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for DenseFPElementsAttr (0.2 sec)

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

        // operand if necessary.
        DenseFPElementsAttr cst;
        if (!matchPattern(binary_op->getOperand(1), m_Constant(&cst)))
          return failure();
        if (cst.getNumElements() != 1) return failure();
        APFloat cst_value = *cst.value_begin<APFloat>();
    
        // Affine op.
        Value filter = fc_op.getFilter();
        Value bias = fc_op.getBias();
        DenseFPElementsAttr filter_cst, bias_cst;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

                                             &data_format)) {
          return rewriter.notifyMatchFailure(
              div_op, "not the root of spatial pooling without dilation");
        }
    
        DenseFPElementsAttr divisor;
        auto div_rhs = recursivelyWalkUp<mhlo::BroadcastInDimOp>(div_op.getRhs());
        if (matchPattern(div_rhs, m_Constant(&divisor))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      auto layer_stats =
          mlir::cast<mlir::DenseFPElementsAttr>(stats_op.getLayerStats());
      std::optional<mlir::ElementsAttr> axis_stats = stats_op.getAxisStats();
      std::optional<uint64_t> axis = stats_op.getAxis();
      std::vector<float> mins, maxs;
      mlir::DenseFPElementsAttr min_max_attr =
          axis_stats.has_value()
              ? mlir::cast<mlir::DenseFPElementsAttr>(axis_stats.value())
              : layer_stats;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    //===----------------------------------------------------------------------===//
    
    OpFoldResult PowOp::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
      auto constant_y = operands[1].dyn_cast_or_null<DenseFPElementsAttr>();
      if (constant_y && constant_y.isSplat()) {
        APFloat y_value = constant_y.getSplatValue<APFloat>();
        auto output_type = getType().cast<ShapedType>();
        if (y_value.isZero() && output_type.hasStaticShape()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top