Searched refs:scatter_dim_input_size (Results 1 – 1 of 1) sorted by relevance
2104 int64_t scatter_dim_input_size = in InferReduceScatterShape() local2106 if (scatter_dim_input_size % shard_count != 0) { in InferReduceScatterShape()2111 scatter_dim_input_size, shard_count); in InferReduceScatterShape()2116 scatter_dim_input_size / shard_count); in InferReduceScatterShape()