Searched refs:packed_input_indices (Results 1 – 2 of 2) sorted by relevance
384 llvm::SmallVector<int64_t, 8> packed_input_indices; in ReplicateCluster() local407 packed_input_indices.push_back(tpu_replicated_input_index); in ReplicateCluster()419 replicated_input_indices.append(packed_input_indices.begin(), in ReplicateCluster()420 packed_input_indices.end()); in ReplicateCluster()
580 packed_input_indices, packed_input_tangents = (589 self.assertAllEqual(expected_indices, packed_input_indices)