Searched refs:output_feature_dim (Results 1 – 5 of 5) sorted by relevance
103 auto output_feature_dim = dnums.output_feature_dimension(); in Run() local111 new_output_dim_order[num_dims - 1] = output_feature_dim; in Run()112 new_conv_dims[num_dims - 1] = hlo->shape().dimensions(output_feature_dim); in Run()
490 auto output_feature_dim = dim_numbers.output_feature_dimension(); in HandleConvolution() local518 conv_slice_shape.set_dimensions(output_feature_dim, filter_slice_width); in HandleConvolution()530 convolution->shape(), sliced_convolutions, output_feature_dim); in HandleConvolution()
446 const int64 output_feature_dim = dnums.output_feature_dimension(); in HandleConvolution() local450 ShapeUtil::GetDimension(result_shape, output_feature_dim); in HandleConvolution()
131 int64 output_feature_dim = filter_shape.dimensions_size() - 1; in ReshapeFilterForDepthwiseConvolution() local132 int64 depthwise_multiplier = filter_shape.dimensions(output_feature_dim); in ReshapeFilterForDepthwiseConvolution()139 output_feature_dim, depthwise_multiplier * input_feature); in ReshapeFilterForDepthwiseConvolution()
125 auto output_feature_dim = conv_dnums.output_feature_dimension(); in MatchBackwardFilter() local246 backward_conv_dnums.set_kernel_output_feature_dimension(output_feature_dim); in MatchBackwardFilter()