Searched refs:output_feature_dim (Results 1 – 4 of 4) sorted by relevance
104 auto output_feature_dim = dnums.output_feature_dimension(); in Run() local112 new_output_dim_order[num_dims - 1] = output_feature_dim; in Run()113 new_conv_dims[num_dims - 1] = hlo->shape().dimensions(output_feature_dim); in Run()
51 int64 output_feature_dim = filter_shape.dimensions_size() - 1; in GroupedFilterShapeForDepthwiseConvolution() local52 int64 depthwise_multiplier = filter_shape.dimensions(output_feature_dim); in GroupedFilterShapeForDepthwiseConvolution()58 grouped_filter_shape.set_dimensions(output_feature_dim, in GroupedFilterShapeForDepthwiseConvolution()
188 auto output_feature_dim = conv_dnums.output_feature_dimension(); in MatchBackwardFilter() local324 backward_conv_dnums.set_kernel_output_feature_dimension(output_feature_dim); in MatchBackwardFilter()
581 const int64 output_feature_dim = dnums.output_feature_dimension(); in HandleConvolution() local585 ShapeUtil::GetDimension(result_shape, output_feature_dim); in HandleConvolution()