Searched defs: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()442 int64_t kernel_output_feature_dim = in HandleConvolution() local
3171 const int64_t kernel_output_feature_dim = in PropagateOnBackpropFilterConv() local
46 auto kernel_output_feature_dim = dnums.kernel_output_feature_dimension(); in Run() local
106 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension(); in MatchBackwardFilter() local
3457 const int64_t kernel_output_feature_dim = in PartitionDot() local3481 const int64_t kernel_output_feature_dim = in PartitionDot() local
996 int64_t kernel_output_feature_dim = num_spatial_dims + 1; in GetConvDimensionNumbersAttr() local