- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for lib_def (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
} tensorflow::FunctionDefLibrary flib; auto flib_def = std::make_unique<tensorflow::FunctionLibraryDefinition>( tensorflow::OpRegistry::Global(), flib); tensorflow::XlaCompiler::Options options; options.device_type = tensorflow::DeviceType(kArbitraryDeviceName); options.client = client; options.flib_def = flib_def.get(); tensorflow::XlaCompiler compiler(options);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc
// Convert MLIR to Graph FunctionLibraryDefinition flib_def(OpRegistry::Global(), FunctionDefLibrary()); GraphExportConfig confs; auto graph = std::make_unique<Graph>(flib_def); absl::flat_hash_set<Node*> control_ret_nodes; Status status = tensorflow::tf2xla::v2::ConvertMlirToGraph( module_in, confs, &graph, &flib_def, &control_ret_nodes); if (!status.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
const ConfigProto& config_proto, const FunctionOptimizationPass::FunctionOptions& function_options, std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, std::vector<std::string>* control_ret_node_names, bool* control_rets_updated) { // overall_state equals to: // Enabled if at least one pass is Enabled. // Disabled if all passes are Disabled.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
"TPU compile op requires module with a entry function main"); } // Export functions to the library. auto flib_def = std::make_unique<FunctionLibraryDefinition>( OpRegistry::Global(), FunctionDefLibrary()); TF_RETURN_IF_ERROR(PrepareAndExportToLibrary(*mlir_module, flib_def.get())); if (VLOG_IS_ON(2)) { tensorflow::DumpMlirOpToFile("legalize_with_old_bridge_post_transform",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.h
Status ConvertMlirToGraph(mlir::ModuleOp module, const GraphExportConfig& configs, std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, absl::flat_hash_set<Node*>* control_ret_nodes); // Converts an MLIR module to TensorFlow graph and FunctionLibraryDefinition.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
*(fdef.add_function()) = my_func; tensorflow::FunctionLibraryDefinition flib_def( tensorflow::OpRegistry::Global(), fdef); OpInputList guaranteed_constants; NameAttrList function; FunctionToHloArgs function_to_hlo_args{&function, &flib_def, /*graph_def_version=*/0,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0)