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

Provide an accessor for dynamic dimension inference.

Browse Source 
PiperOrigin-RevId: 295996157
Change-Id: I50ea04cd692d1163b2e05d9f8e12dbeffc11fa3d
This commit is contained in:
A. Unique TensorFlower 2020-02-19 10:11:55 -08:00 committed by TensorFlower Gardener
parent eea4079931
commit e0575253d1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -133,6 +133,10 @@ class HloEvaluator : public DfsHloVisitorWithDefault {
dynamic_dimension_inference_ = dynamic_dimension_inference;
}
DynamicDimensionInference* dynamic_dimension_inference() {
return dynamic_dimension_inference_;
}
// Enable the fast path for certain operations like dot or convolution.
void set_use_fast_path(bool value) { use_fast_path_ = value; }