Searched refs:IsPartitionedCall (Results 1 – 18 of 18) sorted by relevance
32 return GetFunctionCallInlinePolicy(n->IsPartitionedCall(), found && match); in GetFunctionCallInlinePolicy()
115 if (op->IsPartitionedCall()) partitioned_call_count++; in TEST()185 if (op->IsPartitionedCall()) partitioned_call_count++; in TEST()
68 if (n->IsPartitionedCall()) { in RewriteFunctionCallNode()
887 p.first->IsPartitionedCall() in ExpandInlineFunctions()
68 if (grappler::IsPartitionedCall(node) || in CanonicalizeNodes()
121 } else if (IsPartitionedCall(*input.node_view()->node()) || in FindForwardNode()
85 if (!IsPartitionedCall(func_node) && !IsStatefulPartitionedCall(func_node)) { in IsIndirectFunctionCall()393 if (IsPartitionedCall(node) || IsStatefulPartitionedCall(node)) { in FindFunctionCall()1304 if (n->IsPartitionedCall() || force_inline_as_multi_device) { in InlineFunctionCalls()
135 bool IsPartitionedCall(const NodeDef& node);
418 bool IsPartitionedCall(const NodeDef& node) { in IsPartitionedCall() function
203 if (node.IsPartitionedCall()) { in GetFunctionNameAndAttr()
1225 if (node.IsPartitionedCall()) { in GetPotentialFunctionName()
193 bool IsPartitionedCall() const { return class_ == NC_PARTITIONED_CALL; } in IsPartitionedCall() function
874 if (!n.IsPartitionedCall()) { in BuildXlaLaunchOps()
946 if (op->IsPartitionedCall()) in TEST_F()
2229 } else if (call_node->IsPartitionedCall()) { in LiftOutsideCompilationOnlyArgsFromCallNode()2286 if (call_node->IsPartitionedCall()) { in LiftOutsideCompilationOnlyArgsFromCallNode()2356 if (call_node->IsPartitionedCall()) { in LiftOutsideCompilationOnlyArgs()
605 bool TFGraphDialect::IsPartitionedCall(TFOp op) const { in IsPartitionedCall() function in mlir::tfg::TFGraphDialect
113 bool IsPartitionedCall(TFOp op) const;
1822 } else if (n->IsPartitionedCall()) { in ExtractOutsideCompilationForFuncCallNode()