Searched refs:kernel_output_feature_dim (Results 1 – 6 of 6) sorted by relevance
168 int64_t kernel_output_feature_dim, int64_t group_count, in GetExpandedFilterMask() argument175 int64_t output_feature = filter_shape.dimensions(kernel_output_feature_dim); in GetExpandedFilterMask()191 mask_shape, mask2, {kernel_output_feature_dim})); in GetExpandedFilterMask()442 int64_t kernel_output_feature_dim = in HandleConvolution() local448 kernel_output_feature_dim, group_count, add); in HandleConvolution()453 (group_count == filter->shape().dimensions(kernel_output_feature_dim)); in HandleConvolution()498 filter->shape().dimensions(kernel_output_feature_dim) / group_count; in HandleConvolution()502 if (i == kernel_output_feature_dim) { in HandleConvolution()509 if (kernel_input_feature_dim > kernel_output_feature_dim) { in HandleConvolution()514 if (dim > kernel_output_feature_dim) { in HandleConvolution()[all …]
3171 const int64_t kernel_output_feature_dim = in PropagateOnBackpropFilterConv() local3178 kernel_output_feature_dim); in PropagateOnBackpropFilterConv()
46 auto kernel_output_feature_dim = dnums.kernel_output_feature_dimension(); in Run() local93 new_kernel_dim_order[num_dims - 1] = kernel_output_feature_dim; in Run()95 kernel->shape().dimensions(kernel_output_feature_dim); in Run()
106 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension(); in MatchBackwardFilter() local235 backward_conv_dnums.set_output_feature_dimension(kernel_output_feature_dim); in MatchBackwardFilter()
3457 const int64_t kernel_output_feature_dim = in PartitionDot() local3467 rhs.base_shape().dimensions(kernel_output_feature_dim) || in PartitionDot()3481 const int64_t kernel_output_feature_dim = in PartitionDot() local3485 rhs.base_shape().dimensions(kernel_output_feature_dim) || in PartitionDot()
996 int64_t kernel_output_feature_dim = num_spatial_dims + 1; in GetConvDimensionNumbersAttr() local1006 kernel_input_feature_dim, kernel_output_feature_dim, in GetConvDimensionNumbersAttr()