Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 986 for Auto (0.11 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/strip_saved_module_metadata.cc

    };
    
    bool ShouldStripAttr(NamedAttribute &namedAttr) {
      auto name = namedAttr.getName().strref();
      return name.starts_with("tf_saved_model.");
    }
    
    void StripModule(Operation *module) {
      auto stripAttrs = llvm::to_vector<4>(llvm::make_filter_range(
          module->getAttrs(),
          [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); }));
      for (auto namedAttr : stripAttrs) {
        module->removeAttr(namedAttr.getName());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 08:06:04 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/mark_initialized_variables.cc

      // Get resources from Session.
      auto resource_tensors_or = GetResourcesFromSession(var_ops, session);
      if (!resource_tensors_or.ok())
        return function->emitError(resource_tensors_or.status().message().data());
    
      MLIRContext* context = function.getContext();
      for (auto var_and_tensor : llvm::zip(var_ops, resource_tensors_or.value())) {
        auto& var_op = std::get<0>(var_and_tensor);
        auto& resource_tensor = std::get<1>(var_and_tensor);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 13 19:14:56 UTC 2023
    - 4.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc

        Value new_arg = arg;
        Location loc = func.getLoc();
        if (arg.hasOneUse() && llvm::isa<QuantizeOp>(*arg.user_begin())) {
          auto quantize_op = llvm::cast<QuantizeOp>(*arg.user_begin());
          auto quantize_output = quantize_op.getOutput();
          auto current_type = quant::QuantizedType::getQuantizedElementType(
                                  quantize_output.getType())
                                  .getStorageType();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  4. platforms/jvm/jvm-services/src/test/groovy/org/gradle/jvm/toolchain/internal/AutoInstalledInstallationSupplierTest.groovy

                jdk1.absolutePath,
                jdk2.absolutePath,
                jdk3.absolutePath
            ])
            directories*.source == ["Auto-provisioned by Gradle", "Auto-provisioned by Gradle", "Auto-provisioned by Gradle"]
        }
    
        def "automatically enabled if downloads are enabled"() {
            def jdk = temporaryFolder.createDir("11.0.6.hs-adpt")
    
            given:
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Thu Apr 11 23:01:05 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc

      auto output_fbm = tflite::FlatBufferModel::BuildFromBuffer(
          reinterpret_cast<const char*>(output_builder.GetCurrentBufferPointer()),
          output_builder.GetSize());
      tflite::ModelT output_model;
      output_fbm->GetModel()->UnPackTo(&output_model);
    
      // Extract output metadata
      std::map<std::string, std::string> output_metadata;
      for (const auto& metadata : output_model.metadata) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc

    Operation* GetOpOfValue(Value value) {
      if (auto block_arg = mlir::dyn_cast<BlockArgument>(value))
        return block_arg.getOwner()->getParentOp();
    
      return value.getDefiningOp();
    }
    
    // Check if the TPUCopyWithDynamicShapeOp is valid.
    // 1. The op should be wrapped inside a launch op.
    // 2. The wrapped launch op should be placed on CPU.
    LogicalResult CheckOpIsValid(Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc

      for (int non_bias : non_biases) {
        Operation *non_bias_define = op->getOperand(non_bias).getDefiningOp();
        if (auto dequant = llvm::dyn_cast<TFL::DequantizeOp>(non_bias_define)) {
          auto non_bias_type = mlir::cast<TensorType>(dequant.getInput().getType());
          auto non_bias_ele_type =
              mlir::cast<quant::QuantizedType>(non_bias_type.getElementType());
          non_bias_types.push_back(non_bias_ele_type);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc

                                    PatternRewriter& rewriter) const override {
        if (call_op.getNumResults() != 1) return failure();
        auto users = call_op->getUsers();
        for (auto user : users) {
          if (!llvm::isa<mlir::stablehlo::UniformDequantizeOp>(user)) {
            return failure();
          }
        }
        auto func_name = call_op.getCallee();
        if (!func_name.starts_with("quantized_")) return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc

          shape[0] = ShapedType::kDynamic;
          auto extensions =
              mhlo::TypeExtensionsAttr::get(func->getContext(), bounds);
          auto resultType =
              RankedTensorType::get(shape, inputType.getElementType(), extensions);
          arg.setType(resultType);
        }
        llvm::SmallVector<Type, 8> arg_types;
        for (auto arg : func.getArguments()) arg_types.push_back(arg.getType());
        func.setType(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/utils/arithmetic_count_util.h

                                                               int64_t* count) {
        auto weight = op->getOperand(1);
        auto weight_type =
            mlir::dyn_cast_or_null<mlir::RankedTensorType>(weight.getType());
        if (weight_type == nullptr || !weight_type.hasStaticShape()) return false;
    
        auto output = op->getResult(0);
        auto output_type =
            mlir::dyn_cast_or_null<mlir::RankedTensorType>(output.getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.1K bytes
    - Viewed (0)
Back to top