Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 67 for BoolAttr (0.1 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

        } else {
          bool disable_call_shape_inference = false;
          if (op->hasAttr("_disable_call_shape_inference")) {
            disable_call_shape_inference =
                op->getAttrOfType<BoolAttr>("_disable_call_shape_inference")
                    .getValue();
          }
          inner_op =
              rewriter.create<LegacyCallOp>(loc, new_types, inner_op_operands,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        // Set to `nullptr` because this attribute only matters when the input is
        // dynamic-range quantized.
        const BoolAttr asymmetric_quantize_inputs = nullptr;
    
        const int lhs_rank = lhs_value.getType().cast<TensorType>().getRank();
        const BoolAttr adj_x =
            (lhs_contracting_dims[0] == lhs_rank - 2 ? rewriter.getBoolAttr(true)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

                        const QuantPassSpec& quant_params) {
      rewriter.setInsertionPointAfter(quantized_op);
      FloatAttr tolerance = rewriter.getF32FloatAttr(
          quant_params.numeric_verify_spec.error_tolerance);
      BoolAttr log =
          rewriter.getBoolAttr(quant_params.numeric_verify_spec.log_if_failed_flag);
      // Verify the quantized value by sending the result to the verifier.
      rewriter.create<VerifierT>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

                        Attribute value) {
      ShapedType type;
      if (auto elem_attr = mlir::dyn_cast<ElementsAttr>(value)) {
        return ConstOp::build(builder, result, elem_attr);
      } else if (mlir::isa<BoolAttr, FloatAttr, IntegerAttr>(value)) {
        // All TensorFlow types must be tensor types. In the build() method,
        // we want to provide more flexibility by allowing attributes of scalar
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

    // destination type is not inferred and must be given explicitly.
    //
    // Preconditions: The given value must have a ShapedType.
    static Value CreateTFCastOpF32(OpBuilder *builder, Location loc, Value x,
                                   BoolAttr truncate) {
      auto x_type = mlir::dyn_cast_or_null<ShapedType>(x.getType());
      if (!x_type) llvm_unreachable("unsupported type");
      Type type = x_type.clone(builder->getF32Type());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      auto array = attr.cast<mlir::ArrayAttr>().getValue();
    
      for (int i = 0; i < array.size(); i++) {
        if (llvm::isa<mlir::BoolAttr>(array[i])) {
          flex_builder->Bool(name.c_str(),
                             array[i].cast<mlir::BoolAttr>().getValue());
        } else if (llvm::isa<mlir::StringAttr>(attr)) {
          flex_builder->String(name.c_str(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

      return Unimplemented("SetAttrFloat has not been implemented yet.");
    }
    Status MlirAbstractOp::SetAttrBool(const char* attr_name, bool value) {
      attrs_[attr_name] = BoolAttr::get(context_, value);
      return absl::OkStatus();
    }
    Status MlirAbstractOp::SetAttrShape(const char* attr_name, const int64_t* dims,
                                        const int num_dims) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

    // F32Attr already returns a float as required by flatbuffer builders.
    static float ConvertF32AttrForOptionWriter(
        llvm::APFloat f, flatbuffers::FlatBufferBuilder* builder) {
      return f.convertToFloat();
    }
    
    // BoolAttr already returns a bool as required by flatbuffer builders.
    static bool ConvertBoolAttrForOptionWriter(
        bool b, flatbuffers::FlatBufferBuilder* builder) {
      return b;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

          TF::TensorListLengthOp op, OpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        Location loc = op.getLoc();
        Value input_handle = adaptor.getOperands()[0];
    
        BoolAttr true_attr = rewriter.getBoolAttr(true);
        auto shape = rewriter.create<TF::ShapeOp>(loc, input_handle,
                                                  /*use_32bit=*/true_attr);
        rewriter.replaceOpWithNewOp<TF::GatherOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc

    };
    
    TF::ReshapeOp createOutputReshapeOpForDynamic(
        Value value, ArrayRef<int64_t> shape, Value org_lhs, Value org_rhs,
        EinsumDimensionNumbers& dnums, Location loc, PatternRewriter* rewriter) {
      BoolAttr true_attr = rewriter->getBoolAttr(true);
      // Build ShapeOp
      auto shape_lhs = rewriter->create<TF::ShapeOp>(loc, org_lhs, true_attr);
      auto shape_rhs = rewriter->create<TF::ShapeOp>(loc, org_rhs, true_attr);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.3K bytes
    - Viewed (0)
Back to top