Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for constant_ops (0.13 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc

      auto one_cst =
          rewriter.create<stablehlo::ConstantOp>(rewriter.getUnknownLoc(), one_val);
    
      auto half_val = DenseElementsAttr::get(type, kHalf);
      auto half_cst =
          rewriter.create<stablehlo::ConstantOp>(one_cst.getLoc(), half_val);
    
      auto one_over_root2_val = DenseElementsAttr::get(type, kOneOverRoot2);
      auto one_over_root2_cst = rewriter.create<stablehlo::ConstantOp>(
          half_cst.getLoc(), one_over_root2_val);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 9.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/shlo_simplify.td

    def CloneF32ElementsAttrWithOnes
      : NativeCodeCall<"DenseElementsAttr::get($0.getType().cast<ShapedType>(), (float)1.0)">;
    
    def NotConstant : Constraint<
        CPred<"$0.isa<BlockArgument>() || !llvm::isa<stablehlo::ConstantOp>($0.getDefiningOp())">,
        "Is not a constant.">;
    
    def : Pat<(StableHLO_DivOp $l,
                (StableHLO_ConstantOp:$divisor FloatElementsAttr<32>:$cst)),
              (StableHLO_MulOp $l,
                (StableHLO_DivOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 03:05:20 UTC 2024
    - 1.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.cc

        scale_spec->has_same_operand_and_result_type_requirement = true;
      }
      return scale_spec;
    }
    
    bool IsOpQuantizableStableHlo(Operation* op) {
      if (isa<func::ConstantOp, mlir::stablehlo::ConstantOp>(op)) {
        // Constant ops do not have QuantizableResult attribute but can be
        // quantized.
        return true;
      } else if (op->hasTrait<OpTrait::IsTerminator>() ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 05:56:10 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc

      builder.setInsertionPointToStart(&session_init_func.getBlocks().front());
      auto var_handle_op_in_init = var_handle_op->clone();
      builder.insert(var_handle_op_in_init);
      auto const_op = builder.create<mlir::arith::ConstantOp>(
          session_init_func.getLoc(), tensor_attr.getType(), tensor_attr);
    
      builder.create<TF::AssignVariableOp>(
          session_init_func.getLoc(), llvm::ArrayRef<mlir::Type>{},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc

    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
    
    namespace tflite {
    namespace {
    
    bool IsConst(mlir::Operation* op) {
      return llvm::isa<mlir::arith::ConstantOp, mlir::TF::ConstOp,
                       mlir::TFL::ConstOp, mlir::TFL::QConstOp>(op);
    }
    
    bool IsOpSupported(mlir::Operation* op, const std::string& hardware) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 06:11:34 UTC 2024
    - 7.5K bytes
    - Viewed (0)
Back to top