初始化算子预加载文件示例
初始化算子预加载文件需与MxInitFromConfig接口配合使用。
{ "Operations": [ { "name": "Add", "preload_list": [ { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float;float", "output_shape": "1,3,16,16", "output_type": "float" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float16;float16", "output_shape": "1,3,16,16", "output_type": "float16" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "uint8;uint8", "output_shape": "1,3,16,16", "output_type": "uint8" } ] }, { "name": "Subtract", "preload_list": [ { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float;float", "output_shape": "1,3,16,16", "output_type": "float" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float16;float16", "output_shape": "1,3,16,16", "output_type": "float16" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "uint8;uint8", "output_shape": "1,3,16,16", "output_type": "uint8" } ] }, { "name": "Multiply", "preload_list": [ { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float;float", "output_shape": "1,3,16,16", "output_type": "float" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float16;float16", "output_shape": "1,3,16,16", "output_type": "float16" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "uint8;uint8", "output_shape": "1,3,16,16", "output_type": "uint8" } ] }, { "name": "Divide", "preload_list": [ { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float;float", "output_shape": "1,3,16,16", "output_type": "float" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "float16;float16", "output_shape": "1,3,16,16", "output_type": "float16" }, { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "uint8;uint8", "output_shape": "1,3,16,16", "output_type": "uint8" } ] }, { "name": "BitwiseAnd", "preload_list": [ { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "uint8;uint8", "output_shape": "1,3,16,16", "output_type": "uint8" } ] }, { "name": "BitwiseOr", "preload_list": [ { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "uint8;uint8", "output_shape": "1,3,16,16", "output_type": "uint8" } ] }, { "name": "BitwiseXor", "preload_list": [ { "input_shape": "1,3,16,16;1,3,16,16", "input_type": "uint8;uint8", "output_shape": "1,3,16,16", "output_type": "uint8" } ] }, { "name": "Split", "preload_list": [ { "input_shape": "16,16,3", "input_type": "float", "output_shape": "16,16,1;16,16,1;16,16,1", "output_type": "float;float;float" } ] }, { "name": "Merge", "preload_list": [ { "input_shape": "16,16,1;16,16,2", "input_type": "float;float", "output_shape": "16,16,3", "output_type": "float" } ] }, { "name": "Clip", "preload_list": [ { "input_shape": "1,16,16,3", "input_type": "float", "output_shape": "1,16,16,3", "output_type": "float" }, { "input_shape": "1,16,16,3", "input_type": "float16", "output_shape": "1,16,16,3", "output_type": "float16" }, { "input_shape": "1,16,16,3", "input_type": "uint8", "output_shape": "1,16,16,3", "output_type": "uint8" } ] }, { "name": "Tile", "preload_list": [ { "input_shape": "1,16,16,1", "input_type": "float", "output_shape": "1,16,16,3", "output_type": "float" } ] }, { "name": "Rotate", "preload_list": [ { "input_shape": "5,4,3", "input_type": "uint8", "output_shape": "5,4,3", "output_type": "uint8" } ] }, { "name": "Transpose", "preload_list": [ { "input_shape": "2,3,2", "input_type": "uint8", "output_shape": "2,3,2", "output_type": "uint8" } ] } ] }
父主题: 附录