Searched defs:input_feature_dim (Results 1 – 7 of 7) sorted by relevance
42 auto input_feature_dim = dnums.input_feature_dimension(); in Run() local
50 int64_t input_feature_dim = filter_shape.dimensions_size() - 2; in GroupedFilterShapeForDepthwiseConvolution() local
182 auto input_feature_dim = conv_dnums.input_feature_dimension(); in MatchBackwardFilter() local
104 int64_t input_feature_dim) { in ExpandedFilterShape()
1006 const int64_t input_feature_dim = dnums.input_feature_dimension(); in RewriteDynamicConvolutionForward() local1090 const int64_t input_feature_dim = dnums.input_feature_dimension(); in RewriteDynamicConvolutionKernelGrad() local
581 const int64_t input_feature_dim = dnums.input_feature_dimension(); in HandleConvolution() local
2170 int64_t input_feature_dim = in matchAndRewrite() local