Searched refs:AxesCount (Results 1 – 4 of 4) sorted by relevance
67 if (AxesCount(input_axes_order) == 2) { in ReorderAxes()
274 int AxesCount(AxesOrder axes_order);
2053 CHECK_EQ(AxesCount(input_axes_order), AxesCount(output_axes_order)); in GetShuffleShape()2060 } else if (AxesCount(input_axes_order) == 2) { in GetShuffleShape()2204 int AxesCount(AxesOrder axes_order) { in AxesCount() function
127 CHECK_EQ(AxesCount(output_axes_order), AxesCount(input_axes_order)); in ExportFloatArray()129 CHECK_EQ(input_shape.dimensions_count(), AxesCount(input_axes_order)); in ExportFloatArray()