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