- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for rewrite_subgraph_fn (0.14 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_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)