Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for dump_fetch_nodes (0.18 sec)

  1. tensorflow/compiler/aot/flags.cc

           "is expected to be in the human-readable proto text format, otherwise "
           "it is expected to be in the proto binary format."},
          {"dump_fetch_nodes", &flags->dump_fetch_nodes,
           "If set, only flags related to fetches are processed, and the resulting "
           "fetch nodes will be dumped to stdout in a comma-separated list.  "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 05 16:55:24 UTC 2022
    - 4.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/aot/flags.h

    namespace tfcompile {
    
    // Flags for the tfcompile binary.  See *.cc file for descriptions.
    
    struct MainFlags {
      string graph;
      string debug_info;
      string debug_info_path_begin_marker;
      string config;
      bool dump_fetch_nodes = false;
      string target_triple;
      string target_cpu;
      string target_features;
      string entry_point;
      string cpp_class;
      string out_function_object;
      string out_metadata_object;
      string out_header;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 05 16:55:24 UTC 2022
    - 1.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/aot/compile.cc

      if (flags.config.empty()) {
        return errors::InvalidArgument("Must specify --config");
      }
      TF_RETURN_IF_ERROR(ReadProtoFile(flags.config, &config));
      TF_RETURN_IF_ERROR(ValidateConfig(config));
      if (flags.dump_fetch_nodes) {
        std::set<string> nodes;
        for (const tf2xla::Fetch& fetch : config.fetch()) {
          nodes.insert(fetch.id().node_name());
        }
        std::cout << absl::StrJoin(nodes, ",");
        return absl::OkStatus();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 08:28:57 UTC 2024
    - 11.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/aot/tfcompile.bzl

                cmd = ("CUDA_VISIBLE_DEVICES='' " +
                       "$(location " + tfcompile_tool + ")" +
                       " --config=$(location " + config + ")" +
                       " --dump_fetch_nodes > $@"),
                tools = [tfcompile_tool],
                # Run tfcompile on the build host, rather than forge, since it's
                # typically way faster on the local machine.
                local = 1,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 19:18:08 UTC 2024
    - 21.8K bytes
    - Viewed (0)
Back to top