- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for FunctionDefToBodyHelper (0.34 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/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
function_library->ReachableDefinitions(graph_def).ListFunctionNames()) { const FunctionDef* func_def = function_library->Find(func_name); std::unique_ptr<FunctionBody> func_body; absl::Status status = FunctionDefToBodyHelper( *func_def, AttrSlice(&func_def->attr()), function_library, &func_body); // This is not expected to happen in practice if (!status.ok()) { LOG(ERROR) << "Failed to parse " << func_name << ": "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K 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/jit/extract_outside_compilation_pass.cc
} std::unique_ptr<FunctionBody> then_branch_function_body; TF_RETURN_IF_ERROR(FunctionDefToBodyHelper( *then_branch_function_def, AttrSlice(&then_branch_func.attr()), fld, &then_branch_function_body)); std::unique_ptr<FunctionBody> else_branch_function_body; TF_RETURN_IF_ERROR(FunctionDefToBodyHelper( *else_branch_function_def, AttrSlice(&else_branch_func.attr()), fld,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K 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) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
" in function library."); } std::unique_ptr<FunctionBody> fbody; TF_RETURN_IF_ERROR( FunctionDefToBodyHelper(*fdef, node->attrs(), library, &fbody)); InlineFunctionBodyOptions inline_opts; TF_RETURN_IF_ERROR(InlineFunctionBody(*library, pruned_graph->get(), node,
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/mlir/tensorflow/translate/import_model.cc
tf_name_to_mlir_name_, function_name_uniquifier_, conversion_metadata.function_name); std::unique_ptr<FunctionBody> fbody; TF_RETURN_IF_ERROR( FunctionDefToBodyHelper(*func_def, AttrSlice(), &graph_flib_, &fbody)); TF_RETURN_IF_ERROR(importer.PrepareConvert(*fbody->graph)); TF_ASSIGN_OR_RETURN(auto func_type, importer.InferLibFunctionType(*fbody));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)