Searched refs:GetNumberOfFeatures (Results 1 – 1 of 1) sorted by relevance
85 inline int GetNumberOfFeatures(const TfLiteTensor* input, const int* axis, in GetNumberOfFeatures() function171 int num_features = GetNumberOfFeatures(input, &axis[0], num_axis); in FlexibleLayerNorm()