Searched refs:slicing_dim_size (Results 1 – 1 of 1) sorted by relevance
3234 int64_t slicing_dim_size = in rewriteWithUnknownBegin() local3237 for (int d = slicing_dim_size; d < input_rank; ++d) { in rewriteWithUnknownBegin()3251 SmallVector<int64_t, 4> slice_sizes(slicing_dim_size, 1); in rewriteWithUnknownBegin()3258 for (int d = 0; d < slicing_dim_size; ++d) { in rewriteWithUnknownBegin()3278 for (int d = slicing_dim_size, end = input_shape.size(); d < end; ++d) { in rewriteWithUnknownBegin()