Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for select_user_tf_ops (0.2 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_export.h

      // bool emit_custom_ops          -> toco_flags.allow_custom_ops()
      // bool allow_all_select_tf_ops  -> toco_flags.allow_all_select_tf_ops()
      // std::set<> select_user_tf_ops -> toco_flags.select_user_tf_ops()
      toco::TocoFlags toco_flags;
      // When exporting from SavedModel, this will have the requested tags.
      std::unordered_set<std::string> saved_model_tags;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 09 02:51:43 UTC 2023
    - 2.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.h

    extern llvm::cl::opt<bool> emit_quant_adaptor_ops;
    extern llvm::cl::opt<std::string> quant_stats_file_name;
    extern llvm::cl::opt<bool> convert_tf_while_to_tfl_while;
    extern llvm::cl::opt<std::string> select_user_tf_ops;
    extern llvm::cl::opt<bool> allow_all_select_tf_ops;
    extern llvm::cl::opt<bool> unfold_batchmatmul;
    extern llvm::cl::opt<bool> unfold_large_splat_constant;
    extern llvm::cl::opt<bool> guarantee_all_funcs_one_use;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

      std::unordered_set<std::string> tags;
    
      tensorflow::GraphImportConfig specs;
      specs.upgrade_legacy = upgrade_legacy;
      specs.prune_unused_nodes = true;
    
      if (!select_user_tf_ops.empty() && !emit_select_tf_ops) {
        llvm::errs() << "You must specify `emit-select-tf-ops=true` when passing "
                        "`select-user-tf-ops` flag.";
        return kTrFailure;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 14K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc

    // A list of comma separated TF operators which are created by the user.
    // This must be used with `-emit-select-tf-ops=true`.
    // NOLINTNEXTLINE
    opt<std::string> select_user_tf_ops(
        "select-user-tf-ops",
        llvm::cl::desc(
            "<list of custom tf ops created by the user (comma separated)>"),
        llvm::cl::init(""));
    
    // NOLINTNEXTLINE
    opt<bool> unfold_batchmatmul(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 7.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

            builder_(kInitialBufferSize),
            saved_model_tags_(saved_model_tags),
            allow_all_select_tf_ops_(toco_flags.allow_all_select_tf_ops()),
            select_user_tf_ops_(toco_flags.select_user_tf_ops().begin(),
                                toco_flags.select_user_tf_ops().end()),
            metadata_(metadata),
            supported_backends_(toco_flags.supported_backends().begin(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top