Home
last modified time | relevance | path

Searched refs:keep_caller_node (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dinline_function_utils.cc324 switch (keep_caller_node) { in DebugString()
762 const bool keep_caller_node = in InlineFunctionBody() local
763 options.keep_caller_node == KeepCallerNode::kFetchable || in InlineFunctionBody()
764 options.keep_caller_node == KeepCallerNode::kTargetable; in InlineFunctionBody()
766 if (has_control_outputs || keep_caller_node) { in InlineFunctionBody()
814 if (keep_caller_node) { in InlineFunctionBody()
820 if (options.keep_caller_node == KeepCallerNode::kTargetable || in InlineFunctionBody()
829 } else if (options.keep_caller_node == KeepCallerNode::kFetchable) { in InlineFunctionBody()
Dlower_function_call_op.cc55 inline_options.keep_caller_node = keep_caller_fetchable in RewriteFunctionCallNode()
Dinline_function_utils.h126 KeepCallerNode keep_caller_node = KeepCallerNode::kDoNotKeep; member
Dfunction_test.cc1155 opts.native_options.keep_caller_node = KeepCallerNode::kFetchable; in TEST_F()
1174 opts.native_options.keep_caller_node = KeepCallerNode::kTargetable; in TEST_F()
1240 opts.native_options.keep_caller_node = KeepCallerNode::kDoNotKeep; in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1305 inline_options.keep_caller_node = KeepCallerNode::kFetchable; in InlineFunctionCalls()
1307 inline_options.keep_caller_node = KeepCallerNode::kTargetable; in InlineFunctionCalls()
1309 inline_options.keep_caller_node = KeepCallerNode::kDoNotKeep; in InlineFunctionCalls()