Searched refs:mirrored_variable_indices (Results 1 – 4 of 4) sorted by relevance
388 llvm::SmallVector<int64_t, 8> mirrored_variable_indices; in ReplicateCluster() local416 mirrored_variable_indices.push_back(pos_and_input.index()); in ReplicateCluster()432 if (!mirrored_variable_indices.empty()) in ReplicateCluster()434 builder.getI64ArrayAttr(mirrored_variable_indices)); in ReplicateCluster()
334 std::vector<int> mirrored_variable_indices; in UpdateMirroredVariableIndices() local339 &mirrored_variable_indices)); in UpdateMirroredVariableIndices()342 if (!mirrored_variable_indices.empty()) { in UpdateMirroredVariableIndices()343 for (int i = 0; i < mirrored_variable_indices.size(); ++i) in UpdateMirroredVariableIndices()344 mirrored_variable_indices[i] += additional_per_replica_inputs; in UpdateMirroredVariableIndices()347 mirrored_variable_indices); in UpdateMirroredVariableIndices()765 std::vector<int> mirrored_variable_indices; in RemoveUnusedXlaInput() local770 &mirrored_variable_indices)); in RemoveUnusedXlaInput()919 if (!mirrored_variable_indices.empty()) { in RemoveUnusedXlaInput()922 for (int index : mirrored_variable_indices) { in RemoveUnusedXlaInput()[all …]
229 const std::set<int64>& mirrored_variable_indices() const { in mirrored_variable_indices() function
2394 if (params_info.mirrored_variable_indices().count(i) > 0) { in BuildCompileNode()4030 std::vector<int> mirrored_variable_indices; in GetIOTypes() local4033 &mirrored_variable_indices)); in GetIOTypes()4034 for (int index : mirrored_variable_indices) { in GetIOTypes()