Searched refs:kernel_input_feature_dim (Results 1 – 3 of 3) sorted by relevance
164 const Shape& filter_shape, int64 kernel_input_feature_dim, in GetExpandedFilterMask() argument169 ExpandedFilterShape(filter_shape, group_count, kernel_input_feature_dim); in GetExpandedFilterMask()173 int64 group_size = filter_shape.dimensions(kernel_input_feature_dim); in GetExpandedFilterMask()184 mask_shape, mask1, {kernel_input_feature_dim})); in GetExpandedFilterMask()336 int64 kernel_input_feature_dim = dim_numbers.kernel_input_feature_dimension(); in HandleConvolution() local337 int64 group_size = filter->shape().dimensions(kernel_input_feature_dim); in HandleConvolution()341 kernel_input_feature_dim); in HandleConvolution()343 GetExpandedFilterMask(filter->shape(), kernel_input_feature_dim, in HandleConvolution()359 ShapeUtil::DeleteDimension(kernel_input_feature_dim, filter->shape()); in HandleConvolution()364 if (i == kernel_input_feature_dim) { in HandleConvolution()[all …]
42 auto kernel_input_feature_dim = dnums.kernel_input_feature_dimension(); in Run() local87 new_kernel_dim_order[num_dims - 2] = kernel_input_feature_dim; in Run()89 kernel->shape().dimensions(kernel_input_feature_dim); in Run()
121 auto kernel_input_feature_dim = conv_dnums.kernel_input_feature_dimension(); in MatchBackwardFilter() local235 backward_conv_dnums.set_output_batch_dimension(kernel_input_feature_dim); in MatchBackwardFilter()