- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for rewrite_subgraph_fn (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_subgraphs_pass.h
RewriteSubgraphFn; // Transformation that finds subgraphs whose nodes are marked with // 'group_attribute', splits those subgraphs into functions, and replaces // the originals with function calls. // // 'group_attribute' must be a string valued-attribute that names the new // functions to introduce. // // If 'rewrite_subgraph_fn' is set, it is applied to each subgraph before
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 12 03:59:36 UTC 2022 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
auto flr = pflr->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); 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;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)