OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Relax the folding size threshold to 200 MiB.

Browse Source 
PiperOrigin-RevId: 784254600
This commit is contained in:
Majid Dadashi 2025-07-17 11:33:00 -07:00 committed by TensorFlower Gardener
parent 6be89e0c0a
commit 6f051796d3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
View File

@ -253,7 +253,7 @@ bool ShouldFoldOperation(Operation* inst) {
constexpr int kSizeFactor = 2;
constexpr int64_t kResultsSizeThreshold = (1 << 16); // 64 Kib = 8 KiB
constexpr int64_t kOperandsSizeThreshold = (1 << 30); // 1 Gib = 128 MiB
constexpr int64_t kOperandsSizeThreshold = 200L * 1024 * 1024 * 8; // 200 MiB
return (operands_size <= kOperandsSizeThreshold) &&
((results_size <= kResultsSizeThreshold) ||