Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 29 of 29 for BoolAttr (0.14 sec)

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

        return true;
      }
    
      OpBuilder builder(func);
      bool is_signed = quant_specs_.IsSignedInferenceType();
      IntegerAttr num_bits =
          builder.getI32IntegerAttr(quant_specs_.GetQuantizationTypeWidth());
      BoolAttr narrow_range = builder.getBoolAttr(false);
    
      auto add_quantize_op = [&](Location loc, Type input_type, Block* block,
                                 Block::iterator insertion_point, Value arg,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc

            // the asymmetric_quantize_inputs attribute in the builtin option.
            dynamic_range_quantized_user->setAttr(
                kAsymmetricQuantizeInputsAttr,
                BoolAttr::get(rewriter.getContext(), true));
            changed = true;
          }
        }
        return changed;
      }
    
      // Convert ConstantOp-CastOp-Operation sequence into new ConstantOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.8K bytes
    - Viewed (0)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc

                                  result.attributes))
          return failure();
      }
      bool has_constant = succeeded(parser.parseOptionalKeyword("constant"));
      result.addAttribute("is_constant", BoolAttr::get(context, has_constant));
    
      SmallVector<Type, 1> types;
      if (parser.parseColonTypeList(types)) return failure();
      if (types.size() != 1)
        return parser.emitError(loc) << " expects only a single data type";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 42.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

                                  const Attribute min, const Attribute max,
                                  const int quant_dim, const IntegerAttr num_bits,
                                  const BoolAttr narrow_range, const bool is_signed,
                                  const bool legacy_float_scale,
                                  const bool use_fake_quant_num_bits) {
      SmallVector<double, 4> min_value, max_value;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
Back to top