Searched refs:kernel_output_features (Results 1 – 1 of 1) sorted by relevance
1707 const int64 kernel_output_features = in InferConvolveShape() local1710 if (batch_group_count > 1 && input_batch % kernel_output_features != 0) { in InferConvolveShape()1716 input_batch, kernel_output_features, batch_group_count, in InferConvolveShape()1734 if (kernel_output_features % feature_group_count > 0) { in InferConvolveShape()1746 kernel_output_features, feature_group_count, in InferConvolveShape()1784 dimensions[dnums.output_feature_dimension()] = kernel_output_features; in InferConvolveShape()