- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for peak_loc (0.09 sec)
-
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) -
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)