Searched refs:lhs_concat_dim (Results 1 – 1 of 1) sorted by relevance
139 const HloInstruction* dot_rhs, int64 lhs_concat_dim, in GenNewWindow() argument145 if (lhs_concat_dim != -1) { in GenNewWindow()147 if (conv_dnums.input_spatial_dimensions(i) == lhs_concat_dim) { in GenNewWindow()149 auto lhs_size = dot_lhs->shape().dimensions(lhs_concat_dim + 1); in GenNewWindow()169 lhs_concat_dim == -1) { in GenNewWindow()197 if (lhs_concat_dim != -1) { in GenNewWindow()223 const HloInstruction* dot_rhs, int64 lhs_concat_dim, int64 rhs_concat_dim, in GenNewConvDNums() argument237 if (lhs_concat_dim != -1) { in GenNewConvDNums()238 if (lhs_concat_dim <= input_batch_dimension) { in GenNewConvDNums()241 if (lhs_concat_dim <= input_feature_dimension) { in GenNewConvDNums()[all …]