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

[TF2XLA] Bumping recursion depth limit for conversion

Browse Source 
PiperOrigin-RevId: 351844926
Change-Id: Iac505d5ebb3225be21eb8cb019116b3cb0db9467
This commit is contained in:
George Karpenkov 2021-01-14 11:48:30 -08:00 committed by TensorFlower Gardener
parent 3d21db1f00
commit 8cdf1d3b4d
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/jit/compilability_check_util.h
View File

@ -275,7 +275,7 @@ class RecursiveCompilabilityChecker {
UncompilableNodesMap* uncompilable_nodes_map); UncompilableNodesMap* uncompilable_nodes_map);
// Make sure we don't recurse infinitely on recursive functions. // Make sure we don't recurse infinitely on recursive functions.
const size_t kMaxRecursionDepth = 10; const size_t kMaxRecursionDepth = 50;
const OperationFilter op_filter_; const OperationFilter op_filter_;
const DeviceType jit_device_type_; const DeviceType jit_device_type_;