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

Don't force inlining of functions marked no-inline

Browse Source 
PiperOrigin-RevId: 163842238
This commit is contained in:
Benoit Steiner 2017-08-01 09:42:45 -07:00 committed by TensorFlower Gardener
parent 18718b6f74
commit db0e1c6c8e
1 changed files with 0 additions and 6 deletions

6
tensorflow/core/grappler/grappler_item_builder.cc
View File

@ -83,12 +83,6 @@ Status OptimizeGraph(const GraphDef& graph_def, GraphDef* output_graph_def,
// Inline all functions.
GraphDef inlined_graph_def(graph_def);
for (int i = 0; i < inlined_graph_def.library().function().size(); i++) {
FunctionDef* fdef =
inlined_graph_def.mutable_library()->mutable_function(i);
SetAttrValue(false, &((*fdef->mutable_attr())[kNoInlineAttr]));
}
// Instantiate all variables for function library runtime creation.
std::vector<Device*> devices;
TF_RETURN_IF_ERROR(DeviceFactory::AddDevices(