@@ -322,7 +322,7 @@ class KernelsFactory {
322322 *
323323 * @param input_align [O] Array to be filled with the Input Alignment Restrictions
324324 */
325- virtual void Move_CS_GetInputAlign (uint32_t input_align[kMoveRank ]) {
325+ virtual void Move_CS_GetInputAlign (uint32_t input_align[kMoveRank ], const lib_mli::MoveDataDirection data_dir ) {
326326 set_default_align<kMoveRank >(input_align);
327327 }
328328
@@ -338,7 +338,7 @@ class KernelsFactory {
338338 *
339339 * @param output_align [O] Array to be filled with the Output Alignment Restrictions
340340 */
341- virtual void Move_CS_GetOutputAlign (uint32_t output_align[kMoveRank ]) {
341+ virtual void Move_CS_GetOutputAlign (uint32_t output_align[kMoveRank ], const lib_mli::MoveDataDirection data_dir ) {
342342 set_default_align<kMoveRank >(output_align);
343343 }
344344
@@ -735,7 +735,7 @@ class KernelsFactory {
735735 *
736736 * @param input_align [O] Array to be filled with the Input Alignment Restrictions
737737 */
738- virtual void ClipCS_GetInputAlign (uint32_t input_align[kClipRank ]) {
738+ virtual void Clip_CS_GetInputAlign (uint32_t input_align[kClipRank ]) {
739739 set_default_align<kClipRank >(input_align);
740740 }
741741
@@ -954,7 +954,7 @@ class KernelsFactory {
954954 *
955955 * @param input_align [O] Array to be filled with the Input Alignment Restrictions
956956 */
957- virtual void Mul_GetInputAlign (uint32_t input_align[kEltwiseRank ]) {
957+ virtual void Mul_CS_GetInputAlign (uint32_t input_align[kEltwiseRank ]) {
958958 set_default_align<kEltwiseRank >(input_align);
959959 }
960960
@@ -970,7 +970,7 @@ class KernelsFactory {
970970 *
971971 * @param output_align [O] Array to be filled with the Output Alignment Restrictions
972972 */
973- virtual void Mul_GetOutputAlign (uint32_t output_align[kEltwiseRank ]) {
973+ virtual void Mul_CS_GetOutputAlign (uint32_t output_align[kEltwiseRank ]) {
974974 set_default_align<kEltwiseRank >(output_align);
975975 }
976976
@@ -1100,7 +1100,7 @@ class KernelsFactory {
11001100 *
11011101 * @param input_align [O] Array to be filled with the Input Alignment Restrictions
11021102 */
1103- virtual void Min_GetInputAlign (uint32_t input_align[kEltwiseRank ]) {
1103+ virtual void Min_CS_GetInputAlign (uint32_t input_align[kEltwiseRank ]) {
11041104 set_default_align<kEltwiseRank >(input_align);
11051105 }
11061106
@@ -1116,7 +1116,7 @@ class KernelsFactory {
11161116 *
11171117 * @param output_align [O] Array to be filled with the Output Alignment Restrictions
11181118 */
1119- virtual void Min_GetOutputAlign (uint32_t output_align[kEltwiseRank ]) {
1119+ virtual void Min_CS_GetOutputAlign (uint32_t output_align[kEltwiseRank ]) {
11201120 set_default_align<kEltwiseRank >(output_align);
11211121 }
11221122
0 commit comments