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

Initialize object variables in HLO evaluator.

Browse Source 
The lack of initialization of the dynamic_dimension_inference_ member was causing random crashes when used from a PT client.

PiperOrigin-RevId: 233636726
This commit is contained in:
Davide Libenzi 2019-02-12 10:50:22 -08:00 committed by TensorFlower Gardener
parent 2911f8c92c
commit ce3ee9b6a4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/service/hlo_evaluator.h
View File

@ -331,16 +331,16 @@ class HloEvaluator : public DfsHloVisitorWithDefault {
std::vector<const Literal*> arg_literals_;
// Max loop iterations to execute with no maximum if negative.
int64 max_loop_iterations_;
int64 max_loop_iterations_ = 0;
// Module-level seed handle.
uint64 seed_;
uint64 seed_ = 0;
// RNG engine.
std::minstd_rand0 engine_;
// DynamicDimensionInference is used to evaluate GetDimensionSize, which
// returns the dynamic dimension size of its operand.
DynamicDimensionInference* dynamic_dimension_inference_;
DynamicDimensionInference* dynamic_dimension_inference_ = nullptr;
// Optional handler for custom_call ops.
std::function<StatusOr<Literal>(HloInstruction* custom_call,