Searched refs:UseSpmdForXlaPartitioning (Results 1 – 1 of 1) sorted by relevance
1885 bool UseSpmdForXlaPartitioning(const Node* replicate_node) { in UseSpmdForXlaPartitioning() function1974 const bool use_spmd = (UseSpmdForXlaPartitioning(replicate_node) || in AssignArgsAndRetvalsToCores()2328 UseSpmdForXlaPartitioning(replicate_node) && allow_xla_spmd_partition_ && in BuildCompileNode()