Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for peak_loc (0.09 sec)

  1. tensorflow/compiler/mlir/lite/experimental/remat/rematerializer.cc

                                                     const int peak_loc) const {
      SizeT max_savings = 0;
    
      // We're looking at the outputs of all operators in [`begin`, `end`).  If an
      // output is alive after `peak_loc`, rematerializing the operator anywhere
      // after `peak_loc` would lower the memory profile at `peak_loc`.
      for (int ioperation = begin; ioperation != end; ++ioperation) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 14 20:57:44 UTC 2023
    - 13.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/remat/rematerializer.h

      // [begin, end) after operation `peak_location`
      SizeT MaxSavings(int begin, int end, int peak_loc) const;
    
      // If I want to remat ops [begin, end) after the op at operation `peak_loc`,
      // find the latest point at which to reinsert them (the op before which to
      // insert.)
      int FindBestRematPoint(int begin, int end, int peak_loc) const;
    
      // The memory objects.
      struct Tensor {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 14 20:57:44 UTC 2023
    - 12K bytes
    - Viewed (0)
Back to top