mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-07 00:20:20 +01:00
Don't force inlining of functions marked no-inline
PiperOrigin-RevId: 163842238
This commit is contained in:
parent
18718b6f74
commit
db0e1c6c8e
|
|
@ -83,12 +83,6 @@ Status OptimizeGraph(const GraphDef& graph_def, GraphDef* output_graph_def,
|
||||||
// Inline all functions.
|
// Inline all functions.
|
||||||
GraphDef inlined_graph_def(graph_def);
|
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.
|
// Instantiate all variables for function library runtime creation.
|
||||||
std::vector<Device*> devices;
|
std::vector<Device*> devices;
|
||||||
TF_RETURN_IF_ERROR(DeviceFactory::AddDevices(
|
TF_RETURN_IF_ERROR(DeviceFactory::AddDevices(
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user