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/
Dlower_function_call_op.cc55 inline_options.keep_caller_node = keep_caller_fetchable in RewriteFunctionCallNode()
Dfunction.h257 KeepCallerNode keep_caller_node = KeepCallerNode::kDoNotKeep; member
Dfunction.cc1709 switch (keep_caller_node) { in DebugString()
2125 const bool keep_caller_node = in InlineFunctionBody() local
2126 options.keep_caller_node == KeepCallerNode::kFetchable || in InlineFunctionBody()
2127 options.keep_caller_node == KeepCallerNode::kTargetable; in InlineFunctionBody()
2129 if (has_control_outputs || keep_caller_node) { in InlineFunctionBody()
2178 if (keep_caller_node) { in InlineFunctionBody()
2184 if (options.keep_caller_node == KeepCallerNode::kTargetable || in InlineFunctionBody()
2193 } else if (options.keep_caller_node == KeepCallerNode::kFetchable) { in InlineFunctionBody()
Dfunction_test.cc1028 opts.native_options.keep_caller_node = KeepCallerNode::kFetchable; in TEST_F()
1047 opts.native_options.keep_caller_node = KeepCallerNode::kTargetable; in TEST_F()
1113 opts.native_options.keep_caller_node = KeepCallerNode::kDoNotKeep; in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1271 inline_options.keep_caller_node = KeepCallerNode::kFetchable; in InlineFunctionCalls()
1273 inline_options.keep_caller_node = KeepCallerNode::kTargetable; in InlineFunctionCalls()
1275 inline_options.keep_caller_node = KeepCallerNode::kDoNotKeep; in InlineFunctionCalls()