|
|
|
@ -29,28 +29,7 @@ split_v_op_info = TBERegOp("SplitV") \
|
|
|
|
|
.input(0, "x", False, "required", "all") \
|
|
|
|
|
.output(0, "y", False, "dynamic", "all") \
|
|
|
|
|
.op_pattern("dynamicFormat") \
|
|
|
|
|
.dtype_format(DataType.BOOL_Default, DataType.BOOL_Default) \
|
|
|
|
|
.dtype_format(DataType.BOOL_NHWC, DataType.BOOL_NHWC) \
|
|
|
|
|
.dtype_format(DataType.I8_Default, DataType.I8_Default) \
|
|
|
|
|
.dtype_format(DataType.I8_NHWC, DataType.I8_NHWC) \
|
|
|
|
|
.dtype_format(DataType.U8_Default, DataType.U8_Default) \
|
|
|
|
|
.dtype_format(DataType.U8_NHWC, DataType.U8_NHWC) \
|
|
|
|
|
.dtype_format(DataType.I16_Default, DataType.I16_Default) \
|
|
|
|
|
.dtype_format(DataType.I16_NHWC, DataType.I16_NHWC) \
|
|
|
|
|
.dtype_format(DataType.U16_Default, DataType.U16_Default) \
|
|
|
|
|
.dtype_format(DataType.U16_NHWC, DataType.U16_NHWC) \
|
|
|
|
|
.dtype_format(DataType.I32_Default, DataType.I32_Default) \
|
|
|
|
|
.dtype_format(DataType.I32_NHWC, DataType.I32_NHWC) \
|
|
|
|
|
.dtype_format(DataType.U32_Default, DataType.U32_Default) \
|
|
|
|
|
.dtype_format(DataType.U32_NHWC, DataType.U32_NHWC) \
|
|
|
|
|
.dtype_format(DataType.I64_Default, DataType.I64_Default) \
|
|
|
|
|
.dtype_format(DataType.I64_NHWC, DataType.I64_NHWC) \
|
|
|
|
|
.dtype_format(DataType.U64_Default, DataType.U64_Default) \
|
|
|
|
|
.dtype_format(DataType.U64_NHWC, DataType.U64_NHWC) \
|
|
|
|
|
.dtype_format(DataType.F16_Default, DataType.F16_Default) \
|
|
|
|
|
.dtype_format(DataType.F16_NHWC, DataType.F16_NHWC) \
|
|
|
|
|
.dtype_format(DataType.F32_Default, DataType.F32_Default) \
|
|
|
|
|
.dtype_format(DataType.F32_NHWC, DataType.F32_NHWC) \
|
|
|
|
|
.dtype_format(DataType.None_None, DataType.None_None) \
|
|
|
|
|
.get_op_info()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|