Searched defs:kernel_input_feature_dim (Results 1 – 3 of 3) sorted by relevance
42 auto kernel_input_feature_dim = dnums.kernel_input_feature_dimension(); in Run() local
164 const Shape& filter_shape, int64 kernel_input_feature_dim, in GetExpandedFilterMask()336 int64 kernel_input_feature_dim = dim_numbers.kernel_input_feature_dimension(); in HandleConvolution() local
121 auto kernel_input_feature_dim = conv_dnums.kernel_input_feature_dimension(); in MatchBackwardFilter() local