Searched refs:kernel_output_feature_dim (Results 1 – 5 of 5) sorted by relevance
164 int64 kernel_output_feature_dim, int64 group_count, in GetExpandedFilterMask() argument171 int64 output_feature = filter_shape.dimensions(kernel_output_feature_dim); in GetExpandedFilterMask()187 mask_shape, mask2, {kernel_output_feature_dim})); in GetExpandedFilterMask()436 int64 kernel_output_feature_dim = in HandleConvolution() local442 kernel_output_feature_dim, group_count, add); in HandleConvolution()447 (group_count == filter->shape().dimensions(kernel_output_feature_dim)); in HandleConvolution()492 filter->shape().dimensions(kernel_output_feature_dim) / group_count; in HandleConvolution()496 if (i == kernel_output_feature_dim) { in HandleConvolution()503 if (kernel_input_feature_dim > kernel_output_feature_dim) { in HandleConvolution()508 if (dim > kernel_output_feature_dim) { in HandleConvolution()[all …]
2315 const int64 kernel_output_feature_dim = in PropagateOnBackpropFilterConv() local2322 kernel_output_feature_dim); in PropagateOnBackpropFilterConv()
44 auto kernel_output_feature_dim = dnums.kernel_output_feature_dimension(); in Run() local91 new_kernel_dim_order[num_dims - 1] = kernel_output_feature_dim; in Run()93 kernel->shape().dimensions(kernel_output_feature_dim); in Run()
185 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension(); in MatchBackwardFilter() local314 backward_conv_dnums.set_output_feature_dimension(kernel_output_feature_dim); in MatchBackwardFilter()
1107 IntegerAttr kernel_output_feature_dim = in GetConvDimensionNumbersAttr() local1116 kernel_output_feature_dim, kernel_spatial_dimensions, batch_dim, in GetConvDimensionNumbersAttr()