- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for dump_fetch_nodes (0.18 sec)
-
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) -
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) -
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) -
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)