Searched refs:input_feature_dim (Results 1 – 5 of 5) sorted by relevance
41 auto input_feature_dim = dnums.input_feature_dimension(); in Run() local68 new_input_dim_order[num_dims - 1] = input_feature_dim; in Run()70 input->shape().dimensions(input_feature_dim); in Run()
105 int64 input_feature_dim) { in ExpandedFilterShape() argument110 input_feature_dim, shape.dimensions(input_feature_dim) * group_count); in ExpandedFilterShape()
445 const int64 input_feature_dim = dnums.input_feature_dimension(); in HandleConvolution() local448 ShapeUtil::GetDimension(lhs_shape, input_feature_dim); in HandleConvolution()
130 int64 input_feature_dim = filter_shape.dimensions_size() - 2; in ReshapeFilterForDepthwiseConvolution() local133 int64 input_feature = filter_shape.dimensions(input_feature_dim); in ReshapeFilterForDepthwiseConvolution()137 implicit_broadcast_filter_shape.set_dimensions(input_feature_dim, 1); in ReshapeFilterForDepthwiseConvolution()
119 auto input_feature_dim = conv_dnums.input_feature_dimension(); in MatchBackwardFilter() local230 backward_conv_dnums.set_input_batch_dimension(input_feature_dim); in MatchBackwardFilter()