experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
cherrypick_339370490
STT-tensorflow/tensorflow/python/grappler
History
Saurabh Saxena 5e9107066d Add read variable op to function control outputs 
After inlining, we convert control dependencies to the function call to depend
on all control outputs. Although read variable ops doesn't have side effect
itself, it must be executed before variable updates after the function call.

Note that it's not enough to convert control dependencies to the function call
to depend on both control outputs and data outputs of the function call. The
read variable ops can be inputs to ops that have side effects, e.g. assert and
print, which are not the function data outputs.

PiperOrigin-RevId: 338391537
Change-Id: Ibc843c1a584088e54010685ebce678b047f7d94d
2020-10-21 20:28:26 -07:00
..
arithmetic_optimizer_test.py Improve shape inference in TF 2.0, by propagating input shapes to functions. 2020-01-28 12:39:56 -08:00
auto_mixed_precision_test.py Merge pull request from Intel-tensorflow:nhasabni/auto_mp_ut_fix 2020-09-25 05:30:43 -07:00
cluster_test.py Update grappler/cluster_test.py reflecting changes in random_uniform op 2019-12-04 14:26:25 -08:00
cluster_wrapper.cc Updated code to be compatible with passing string_view 2020-08-20 10:28:22 -07:00
cluster.py Export grappler code from C++ to Python with pybind11 instead of swig. Adding pywrap_required_hdrs to downstream grappler build files which will go away with bazel first class shared library support. 2020-02-03 13:13:08 -08:00
constant_folding_test.py Add read variable op to function control outputs 2020-10-21 20:28:26 -07:00
cost_analyzer_test.py
cost_analyzer_tool.py
cost_analyzer_wrapper.cc Updated code to be compatible with passing string_view 2020-08-20 10:28:22 -07:00
cost_analyzer.cc minor spelling tweaks 2020-02-11 15:09:21 +09:00
cost_analyzer.h
cost_analyzer.py Export grappler code from C++ to Python with pybind11 instead of swig. Adding pywrap_required_hdrs to downstream grappler build files which will go away with bazel first class shared library support. 2020-02-03 13:13:08 -08:00
datasets_test.py
graph_analyzer_tool_wrapper.cc systemlibs: unbundle pybind11 2020-04-08 03:38:38 +08:00
graph_analyzer.py Export graph_analyzer functions from C++ to Python with pybind11 instead of swig. 2019-09-23 12:59:39 -07:00
item_test.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
item_wrapper.cc Updated code to be compatible with passing string_view 2020-08-20 10:28:22 -07:00
item.py Export grappler code from C++ to Python with pybind11 instead of swig. Adding pywrap_required_hdrs to downstream grappler build files which will go away with bazel first class shared library support. 2020-02-03 13:13:08 -08:00
layout_optimizer_test.py Roll forward of https://github.com/tensorflow/tensorflow/pull/42970 with fix for 5D tensors and unsupported axis that cannot use fused batch norm. 2020-10-02 12:08:15 -07:00
memory_optimizer_test.py [Grappler] Replace redundant variable updates (e.g. AssignAdd of all zero deltas) with NoOps or Identity. Also relax a constraint in dependency optimizer to allow pruning of unused variable reads. 2020-06-30 12:13:27 -07:00
model_analyzer_test.py
model_analyzer_wrapper.cc Updated code to be compatible with passing string_view 2020-08-20 10:28:22 -07:00
model_analyzer.cc Sign compare warning fixes batch 2 2020-06-16 03:51:57 +00:00
model_analyzer.h
model_analyzer.py Export model_analyzer functions from C++ to Python with pybind11 instead of swig. 2019-11-20 10:41:58 -08:00
tf_optimizer_test.py
tf_optimizer_wrapper.cc Updated code to be compatible with passing string_view 2020-08-20 10:28:22 -07:00
tf_optimizer.py Export grappler code from C++ to Python with pybind11 instead of swig. Adding pywrap_required_hdrs to downstream grappler build files which will go away with bazel first class shared library support. 2020-02-03 13:13:08 -08:00