Searched refs:replicated_input_indices (Results 1 – 1 of 1) sorted by relevance
383 llvm::SmallVector<int64_t, 8> replicated_input_indices; in ReplicateCluster() local411 replicated_input_indices.push_back(tpu_replicated_input_index); in ReplicateCluster()419 replicated_input_indices.append(packed_input_indices.begin(), in ReplicateCluster()430 builder.getI64ArrayAttr(replicated_input_indices)); in ReplicateCluster()