Searched defs:mirrored_variable_indices (Results 1 – 4 of 4) sorted by relevance
229 const std::set<int64>& mirrored_variable_indices() const { in mirrored_variable_indices() function
334 std::vector<int> mirrored_variable_indices; in UpdateMirroredVariableIndices() local765 std::vector<int> mirrored_variable_indices; in RemoveUnusedXlaInput() local2625 std::vector<int> mirrored_variable_indices; in BuildTPUReplicateOps() local
4030 std::vector<int> mirrored_variable_indices; in GetIOTypes() local
388 llvm::SmallVector<int64_t, 8> mirrored_variable_indices; in ReplicateCluster() local