Searched refs:UseSpmdForXlaPartitioning (Results 1 – 1 of 1) sorted by relevance
2098 bool UseSpmdForXlaPartitioning(const Node* replicate_node) { in UseSpmdForXlaPartitioning() function2187 const bool use_spmd = (UseSpmdForXlaPartitioning(replicate_node) || in AssignArgsAndRetvalsToCores()2622 UseSpmdForXlaPartitioning(replicate_node) && allow_xla_spmd_partition_; in BuildCompileNode()3271 UseSpmdForXlaPartitioning(&replicate_node) && allow_xla_spmd_partition_; in BuildExecuteNodes()