Function GetAxisIndex

tensorflow/lite/delegates/gpu/common/shape.cc:103–105

}

int GetAxisIndex(Layout layout, Axis axis) {
  return DispatchByLayout(layout, GetIndexByAxisFunc{axis});
}

int Size(Layout layout) { return DispatchByLayout(layout, NumAxisFunc()); }