Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 66 for getNumArguments (0.27 sec)

  1. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

      // Relevant argument characteristics are checked by the TFL op definition.
      if (func_.getNumArguments() < 5) {
        return func_.emitWarning()
               << "Invalid number of arguments to "
                  "non_max_suppression_padded_v2 (need at least 5): "
               << func_.getNumArguments();
      }
      if (func_.getFunctionType().getNumResults() != 2) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/freeze_saved_model_assets.cc

        return;
      }
      SymbolTable symbol_table(module);
    
      for (auto func : module.getOps<func::FuncOp>()) {
        llvm::BitVector args_to_erase(func.getNumArguments());
        OpBuilder builder(func.getBody());
    
        for (int i = 0, e = func.getNumArguments(); i < e; ++i) {
          SmallVector<TF::InitializeTableFromTextFileV2Op, 4>
              init_table_from_text_file_ops_to_erase;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jan 30 01:12:09 UTC 2023
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc

      void RewriteFunction(const SymbolTable& symbol_table, func::FuncOp func) {
        if (func.getNumArguments() == 0) {
          return;
        }
    
        auto builder = OpBuilder::atBlockBegin(&func.front());
    
        llvm::SmallDenseMap<llvm::StringRef, TF::ConstOp> const_ops;
        llvm::BitVector arg_indexes_to_remove(func.getNumArguments());
    
        // Replace arguments with const ops.
        for (BlockArgument argument : func.getArguments()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/deduplicate_bound_input_bindings.cc

      if (!mlir::tf_saved_model::IsExported(func)) return;
      llvm::SmallDenseMap<Attribute, unsigned, 8> unique_bound_inputs;
      llvm::BitVector arg_indices_to_erase(func.getNumArguments());
      for (unsigned i = 0, e = func.getNumArguments(); i < e; i++) {
        auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>(
            i, "tf_saved_model.bound_input");
        if (!attr) continue;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 09:25:35 UTC 2022
    - 2.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

    }
    
    LogicalResult ConvertMaxUnpoolingFunc::VerifySignature() {
      // Verify high-level function signature.
      if (func_.getNumArguments() != 2) {
        return func_.emitWarning()
               << "Invalid number of arguments to " << kMaxUnpooling << ": "
               << func_.getNumArguments();
      }
      if (func_.getFunctionType().getNumResults() != 1) {
        return func_.emitWarning()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/optimize_global_tensors.cc

        global_tensor.erase();
      }
    }
    
    void EraseUnusedBoundInputs(ModuleOp module) {
      for (auto func : module.getOps<func::FuncOp>()) {
        llvm::BitVector args_to_erase(func.getNumArguments());
        for (int i = 0, e = func.getNumArguments(); i < e; i++) {
          if (func.getArgAttr(i, "tf_saved_model.bound_input") &&
              func.getArgument(i).use_empty()) {
            args_to_erase.set(i);
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

      if (input_names.size() != main_func.getNumArguments() ||
          output_names.size() != main_func.getNumResults()) {
        module_op.emitError()
            << "Number of inputs and outputs in the tf.entry_function attribute "
               "mismatched. [Input] Expected: "
            << input_names.size() << ", got: " << main_func.getNumArguments()
            << ". [Output] Expected: " << output_names.size()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/materialize_mlir_passthrough_op.cc

          return;
        }
        if (main.getNumArguments() != op->getNumOperands()) {
          op->emitError() << "mismatch between MLIR Opaque Op number of operands ("
                          << op->getNumOperands()
                          << ") and main() entry point in the module ("
                          << main.getNumArguments() << " args)\n";
          return;
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 4.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc

      if (!input_shapes_attr) return success();
    
      if (input_shapes_attr.size() != func_op.getNumArguments()) {
        return func_op->emitError(
                   "Number of arguments and 'tf._input_shapes' "
                   "attribute size do not match. ")
               << "Num args: " << func_op.getNumArguments()
               << ", tf._input_shapes size: " << input_shapes_attr.size();
      }
    
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 09:56:53 UTC 2024
    - 19.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc

      if (!return_op)
        return function.emitError() << "expects function '" << function.getName()
                                    << "' to have a MLIR ReturnOp";
    
      llvm::SmallVector<ResourceInfo, 4> resources(function.getNumArguments());
      auto argument_types =
          llvm::to_vector<4>(function.getFunctionType().getInputs());
      bool has_resources = false;
      auto add_resource_argument = [&](BlockArgument arg,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.1K bytes
    - Viewed (0)
Back to top