Searched refs:kernel_output_features (Results 1 – 1 of 1) sorted by relevance
1747 const int64 kernel_output_features = in InferConvolveShape() local1750 if (kernel_output_features % batch_group_count != 0) { in InferConvolveShape()1755 kernel_output_features, batch_group_count, ShapeUtil::HumanString(lhs), in InferConvolveShape()1772 if (kernel_output_features % feature_group_count > 0) { in InferConvolveShape()1784 kernel_output_features, feature_group_count, in InferConvolveShape()1822 dimensions[dnums.output_feature_dimension()] = kernel_output_features; in InferConvolveShape()