Searched refs:kernel_output_feature_dim (Results 1 – 3 of 3) sorted by relevance
165 int64 kernel_output_feature_dim, int64 group_count, in GetExpandedFilterMask() argument172 int64 output_feature = filter_shape.dimensions(kernel_output_feature_dim); in GetExpandedFilterMask()188 mask_shape, mask2, {kernel_output_feature_dim})); in GetExpandedFilterMask()338 int64 kernel_output_feature_dim = in HandleConvolution() local344 kernel_output_feature_dim, group_count, add); in HandleConvolution()349 (group_count == filter->shape().dimensions(kernel_output_feature_dim)); in HandleConvolution()390 filter->shape().dimensions(kernel_output_feature_dim); in HandleConvolution()489 filter->shape().dimensions(kernel_output_feature_dim); in HandleConvolution()497 filter_slice_starts[kernel_output_feature_dim] = i * filter_slice_width; in HandleConvolution()498 filter_slice_limits[kernel_output_feature_dim] = in HandleConvolution()[all …]
43 auto kernel_output_feature_dim = dnums.kernel_output_feature_dimension(); in Run() local90 new_kernel_dim_order[num_dims - 1] = kernel_output_feature_dim; in Run()92 kernel->shape().dimensions(kernel_output_feature_dim); in Run()
122 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension(); in MatchBackwardFilter() local236 backward_conv_dnums.set_output_feature_dimension(kernel_output_feature_dim); in MatchBackwardFilter()