Searched refs:input_sharding_specs (Results 1 – 1 of 1) sorted by relevance
382 auto input_sharding_specs = py::cast<std::vector<ShardingSpec>>( in PopulateCacheEntry() local389 CHECK_EQ(input_sharding_specs.size(), input_indices.size()); in PopulateCacheEntry()390 cache_entry.input_specs.reserve(input_sharding_specs.size()); in PopulateCacheEntry()391 for (int i = 0; i < input_sharding_specs.size(); ++i) { in PopulateCacheEntry()392 cache_entry.input_specs.emplace_back(input_sharding_specs[i], in PopulateCacheEntry()