Searched refs:input_num_shards (Results 1 – 1 of 1) sorted by relevance
168 const auto input_num_shards = input_layout.num_shards(); in HandleConv() local223 int input_size = input_local_size * input_num_shards[curr_input_dim]; in HandleConv()257 halo_size / (input_num_shards[curr_input_dim] - 1); in HandleConv()