Searched refs:rhs_concat_dim (Results 1 – 1 of 1) sorted by relevance
140 int64 rhs_concat_dim, bool windowed_at_contracting_dims, in GenNewWindow() argument165 if (rhs_concat_dim != -1) { in GenNewWindow()167 if (conv_dnums.kernel_spatial_dimensions(i) == rhs_concat_dim && in GenNewWindow()171 auto rhs_size = dot_rhs->shape().dimensions(rhs_concat_dim + 1); in GenNewWindow()206 if (rhs_concat_dim != -1) { in GenNewWindow()223 const HloInstruction* dot_rhs, int64 lhs_concat_dim, int64 rhs_concat_dim, in GenNewConvDNums() argument251 if (rhs_concat_dim != -1 && !windowed_at_contracting_dims && in GenNewConvDNums()262 if (rhs_concat_dim != -1) { in GenNewConvDNums()263 if (rhs_concat_dim <= kernel_input_feature_dimension) { in GenNewConvDNums()266 if (rhs_concat_dim <= kernel_output_feature_dimension) { in GenNewConvDNums()[all …]