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

[XLA] Initialize fields of RematerializationSizes by default

Browse Source 
PiperOrigin-RevId: 355954409
Change-Id: I0da0d3ce320c53321778ad66ab9703307bc231c7
This commit is contained in:
David Majnemer 2021-02-05 16:41:06 -08:00 committed by TensorFlower Gardener
parent cf18ba2332
commit 7cd52d03c4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/service/hlo_rematerialization.h
View File

@ -45,8 +45,8 @@ class HloRematerialization : public HloModulePass {
// Helper struct that communicates the before / after sizes for the
// rematerialization process.
struct RematerializationSizes {
int64 before_bytes;
int64 after_bytes;
int64 before_bytes = -1;
int64 after_bytes = -1;
};
// Mode in which the rematerialization algorithm should be run.