- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for FunctionDefToBodyHelper (0.24 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
host_compute_core, &fld, &shape_inference_graphs, &has_outside_compilation)); // Get rewritten XLA computation function. std::unique_ptr<FunctionBody> xla_fbody; TF_CHECK_OK(FunctionDefToBodyHelper(*fld.Find("cluster_rewritten"), AttrSlice(), &fld, &xla_fbody)); auto node_name_index = xla_fbody->graph->BuildNodeNameIndex(); // Check XlaHostCompute nodes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
TF_CHECK_OK(RearrangeFunctionArguments( [&](const NameAttrList &function, const FunctionBody **fbody) { std::unique_ptr<FunctionBody> new_fbody; TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*fld.Find(function.name()), AttrSlice(&function.attr()), &fld, &new_fbody)); *fbody = new_fbody.get();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 19.3K bytes - Viewed (0)