Searched defs:replica_index (Results 1 – 2 of 2) sorted by relevance
47 auto create_flattened_id = [&](HloInstruction *replica_index) { in CreateStartIndicesForCollectiveDecomposition()
3919 int replica_index, const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationNodes()