Searched refs:AxesCount (Results 1 – 4 of 4) sorted by relevance
67 if (AxesCount(input_axes_order) == 2) { in ReorderAxes()
276 int AxesCount(AxesOrder axes_order);
2009 CHECK_EQ(AxesCount(input_axes_order), AxesCount(output_axes_order)); in GetShuffleShape()2016 } else if (AxesCount(input_axes_order) == 2) { in GetShuffleShape()2160 int AxesCount(AxesOrder axes_order) { in AxesCount() function
124 CHECK_EQ(AxesCount(output_axes_order), AxesCount(input_axes_order)); in ExportFloatArray()126 CHECK_EQ(input_shape.dimensions_count(), AxesCount(input_axes_order)); in ExportFloatArray()