Searched defs:input_feature_dim (Results 1 – 8 of 8) sorted by relevance
44 auto input_feature_dim = dnums.input_feature_dimension(); in Run() local
47 int64_t input_feature_dim; member
103 auto input_feature_dim = conv_dnums.input_feature_dimension(); in MatchBackwardFilter() local
50 int64_t input_feature_dim = filter_shape.dimensions_size() - 2; in GroupedFilterShapeForDepthwiseConvolution() local
107 int64_t input_feature_dim) { in ExpandedFilterShape()
1063 const int64_t input_feature_dim = dnums.input_feature_dimension(); in RewriteDynamicConvolutionForward() local1146 const int64_t input_feature_dim = dnums.input_feature_dimension(); in RewriteDynamicConvolutionKernelGrad() local
598 const int64_t input_feature_dim = dnums.input_feature_dimension(); in GetConvolutionFlops() local
3454 const int64_t input_feature_dim = in PartitionDot() local