Searched refs:FunctionCallInlinePolicy (Results 1 – 4 of 4) sorted by relevance
20 FunctionCallInlinePolicy GetFunctionCallInlinePolicy( in GetFunctionCallInlinePolicy()23 return FunctionCallInlinePolicy::kMultiDevicePlacer; in GetFunctionCallInlinePolicy()24 return FunctionCallInlinePolicy::kSingleDevicePlacer; in GetFunctionCallInlinePolicy()27 FunctionCallInlinePolicy GetFunctionCallInlinePolicy(const Node* n) { in GetFunctionCallInlinePolicy()
24 enum class FunctionCallInlinePolicy { enum51 FunctionCallInlinePolicy GetFunctionCallInlinePolicy(const Node* n);55 FunctionCallInlinePolicy GetFunctionCallInlinePolicy(
46 FunctionCallInlinePolicy policy = GetFunctionCallInlinePolicy(n); in RewriteFunctionCallNode()47 if (policy == FunctionCallInlinePolicy::kMultiDevicePlacer) { in RewriteFunctionCallNode()56 } else if (policy == FunctionCallInlinePolicy::kSingleDevicePlacer) { in RewriteFunctionCallNode()
208 tensorflow::FunctionCallInlinePolicy policy = in processInlinedCallBlocks()213 if (policy == tensorflow::FunctionCallInlinePolicy::kMultiDevicePlacer) in processInlinedCallBlocks()