- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for forward_tensor (0.15 sec)
-
tensorflow/compiler/mlir/lite/experimental/remat/rematerializer_test.cc
explicit MlpRemat(const std::vector<int>& sizes) { int forward_tensor = -1; int backward_tensor = -1; int op = -1; // Forward pass: for (const int size : sizes) { op = AddOperation(/*is_stateful=*/false); if (forward_tensor >= 0) AddUse(op, forward_tensor); forward_tensor = AddTensor(size); AddUse(op, forward_tensor); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 14 20:57:44 UTC 2023 - 19.1K bytes - Viewed (0)