Searched defs:sizes_i (Results 1 – 3 of 3) sorted by relevance
52 const int* sizes_i = output_sizes + batch_id * output_dim; in remove_padding_transform0213_2() local93 const int* sizes_i = output_sizes + batch_id * output_dim; in remove_padding_2() local126 const int* sizes_i = output_sizes + batch_id * output_dim; in remove_padding() local261 const int* sizes_i = input_sizes + batch_id * input_dim; in add_padding_1() local300 const int* sizes_i = input_sizes + batch_id * input_dim; in add_padding_2() local345 const int* sizes_i = input_sizes + batch_id * input_dim; in add_padding_3() local
223 const auto* sizes_i = input_sizes + b * input_dim; in transform_bias_rescale_qkv_add_padding_kernel() local
233 IntArrayRef sizes_i( in nested_from_padded_generic() local