- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for reuse_existing_functions (0.23 sec)
-
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
// If 'reuse_existing_functions' is set, use an existing function with the // same name, if any. // If 'rewrite_subgraph_fn' is set, it is applied to each subgraph before // function conversion. Status BuildFunctionDefs(const RewriteSubgraphFn& rewrite_subgraph_fn, bool reuse_existing_functions, FunctionLibraryDefinition* library);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
auto output = std::make_unique<Graph>((*graph)->op_registry()); TF_RETURN_WITH_CONTEXT_IF_ERROR( EncapsulateSubgraphsInFunctions( kXlaClusterIdAttr, **graph, RewriteSubgraph, /*reuse_existing_functions=*/true, &output, flib_def), "EncapsulateXlaComputationsPass failed"); graph->swap(output); return absl::OkStatus(); } /*static*/ Status EncapsulateXlaComputationsPass::BuildXlaLaunchOps(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
std::unique_ptr<Graph> graph_out; s = EncapsulateSubgraphsInFunctions("_encapsulate", *graph, /*rewrite_subgraph_fn=*/{}, /*reuse_existing_functions=*/false, &graph_out, lib_def.get()); if (!s.ok()) return s; std::unordered_map<string, XlaClusterInfo> clusters; for (const auto& func : encapsulated_functions) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
new_func_name); TF_RETURN_IF_ERROR(EncapsulateSubgraphsInFunctions( outside_compilation_attr_name, *fbody->graph, *rewrite_fn, /*reuse_existing_functions=*/true, &graph_out, fld)); // Replace outside_compilation function nodes with HostCompute ops. std::vector<Node*> outside_compilation_nodes; for (Node* n : graph_out->nodes()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)