Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 69 of 69 for ConstantOp (0.24 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc

      }
    
      auto scalar_type = RankedTensorType::get(/*shape=*/{}, fp_type);
      auto epsilon_tensor_attr = DenseElementsAttr::get(
          scalar_type, {mlir::cast<Attribute>(epsilon_attr)});
      Value epsilon = b.create<mhlo::ConstantOp>(epsilon_tensor_attr);
      auto dims_type = RankedTensorType::get(/*shape=*/{0}, b.getIntegerType(64));
      auto dims = DenseIntElementsAttr::get(dims_type, SmallVector<int64_t, 1>{});
      if (broadcast_to_type.hasStaticShape()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc

        auto new_shape_const_attr =
            DenseElementsAttr::get(shape_spec_type, new_shape.getShape());
        rewriter.setInsertionPointAfter(weight_op);
        auto new_shape_const = rewriter.create<arith::ConstantOp>(
            weight_op->getLoc(), shape_spec_type, new_shape_const_attr);
        auto reshape_op = rewriter.create<TF::ReshapeOp>(
            weight_op->getLoc(), new_shape, weight_op->getResult(0),
            new_shape_const);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc

      ASSERT_TRUE(module_op);
    
      auto test_func = module_op->lookupSymbol<func::FuncOp>("constant_add");
      ASSERT_THAT(test_func, NotNull());
    
      auto constant_op =
          FindOperationOfType<mlir::stablehlo::ConstantOp>(test_func);
      EXPECT_TRUE(IsOpQuantizableStableHlo(constant_op));
    }
    
    TEST_F(IsOpQuantizableStableHloTest, TerminatorOpNotQuantizable) {
      OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleConstantAdd);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 04 07:19:09 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

          scales, zero_points, quant_dim, qtype.getStorageTypeMin(),
          qtype.getStorageTypeMax());
    }
    
    }  // namespace
    
    bool IsOpQuantizable(Operation* op) {
      if (isa<func::ConstantOp, arith::ConstantOp, quantfork::StatisticsOp>(op)) {
        // Constant ops do not have QuantizableResult attribute but they can deal
        // with quantized tensors.
        return true;
      } else if (op->hasTrait<OpTrait::IsTerminator>() ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

    }
    
    static bool IsConst(Operation* op) {
      return isa<mlir::func::ConstantOp, mlir::arith::ConstantOp, mlir::TF::ConstOp,
                 tfl::ConstOp, tfl::QConstOp, tfl::SparseConstOp,
                 tfl::SparseQConstOp, mlir::TFL::NoValueOp,
                 mlir::stablehlo::ConstantOp, mlir::vhlo::ConstantOpV1>(op);
    }
    
    static bool IsTFResourceOp(Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

      }
    
      auto dense_attr_or = GetDenseAttrFromTensorProtoAttr(
          tensor_proto_attr.getValue(), new_operand_type);
      if (failed(dense_attr_or)) return failure();
    
      return Value(rewriter.create<mhlo::ConstantOp>(op->getLoc(), new_operand_type,
                                                     *dense_attr_or));
    }
    
    xla::ConvolutionDimensionNumbers ConvertConvolutionDimensionNumbers(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc

          // Use mhlo.Constant when it is consumed by the lowering passes since they
          // can't lower tf.Const.
          Value cst;
          if (use_mhlo_const) {
            cst = builder.create<mhlo::ConstantOp>(func_op->getLoc(), attrs);
          } else {
            cst = builder.create<TF::ConstOp>(func_op->getLoc(), attrs);
          }
          func_op.getArgument(i).replaceAllUsesWith(cst);
        }
        return module_op;
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 03 01:03:21 UTC 2024
    - 35.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      LogicalResult matchAndRewrite(ConstOp const_op,
                                    PatternRewriter& rewriter) const override {
        if (arith::ConstantOp::isBuildableWith(const_op.getValue(),
                                               const_op.getType())) {
          rewriter.replaceOpWithNewOp<arith::ConstantOp>(const_op,
                                                         const_op.getValue());
          return success();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

                DenseFPElementsAttr attr;
                if (!matchPattern(q.getOperand(), m_Constant(&attr))) {
                  continue;
                }
                auto cst = rewriter.create<arith::ConstantOp>(
                    quantized_op->getLoc(), attr);
                quantizing_op->setOperand(i, cst.getResult());
              }
            }
    
            for (int i = 0, e = quantized_op->getNumResults(); i < e; ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
Back to top