Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for lib_def (0.1 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top