Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for DecomposeGraph (0.35 sec)

  1. tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.h

      // FunctionDef.
      absl::StatusOr<FunctionDef> ExpandNode(const NodeDef& node_def,
                                             StringPiece func_name);
    
      // Runs the decompose passes on the user_module.
      Status DecomposeGraph(mlir::ModuleOp user_module);
    
      // Erases the tfr_module created.
      void Destroy();
    
     private:
      mlir::ModuleOp tfr_module_;
      mlir::PassManager pm_;
    
      GraphExportConfig export_confs_;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 11:12:54 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc

      // Run the decompose passes on the module
      TF_RETURN_IF_ERROR(DecomposeGraph(module));
    
      // Export the result as a FunctionDef.
      FunctionDef func_def;
      TF_RETURN_IF_ERROR(
          ConvertMlirFunctionToFunctionLibraryDef(func, export_confs_, &func_def));
      module.erase();
      return func_def;
    }
    
    Status TFRDecomposeContext::DecomposeGraph(mlir::ModuleOp user_module) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 29 02:34:43 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfr/integration/graph_decompose_pass.cc

        return absl::OkStatus();
      }
    
      tf_core_op_expansion_graph_counter->GetCell()->IncrementBy(1);
    
      LOG_FIRST_N(INFO, 1) << "Run Graph Decomposition Passes";
    
      TF_RETURN_IF_ERROR(DecomposeGraph(module));
    
      LOG_FIRST_N(INFO, 1) << "Finish Graph Decomposition Passes";
    
      return absl::OkStatus();
    }
    
    namespace {
    constexpr int kMlirGraphDecomposePassPriority = -1;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Feb 25 16:22:36 UTC 2024
    - 2.8K bytes
    - Viewed (0)
Back to top