experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Bump the constant folding threshold up

Browse Source 
PiperOrigin-RevId: 246083525
This commit is contained in:
David Majnemer 2019-04-30 22:48:04 -07:00 committed by TensorFlower Gardener
parent a10615b14f
commit bf8e40b92f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/hlo_constant_folding.cc
View File

@ -130,7 +130,7 @@ StatusOr<bool> HloConstantFolding::Run(HloModule* module) {
int64 elements_in_constant = int64 elements_in_constant =
ShapeUtil::ElementsIn(instruction->shape()); ShapeUtil::ElementsIn(instruction->shape());
static const int64 kMaximumConstantSizeElements = 2 * 1000 * 1000; static const int64 kMaximumConstantSizeElements = 45 * 1000 * 1000;
if (elements_in_constant > elements_in_removed_operands && if (elements_in_constant > elements_in_removed_operands &&
elements_in_constant > kMaximumConstantSizeElements) { elements_in_constant > kMaximumConstantSizeElements) {
continue; continue;