From: @looop5 Reviewed-by: @dylangeng,@anyrenwei Signed-off-by: @anyrenweitags/v1.2.0
| @@ -63,21 +63,36 @@ link_directories(${AKG_RPATH}) | |||
| # Search AKG_EXTEND by order | |||
| set(AKG_EXTEND ) | |||
| if(NOT USE_CUDA) | |||
| set(AKG_EXTEND_FILE ${AKG_SOURCE_DIR}/libakg_ext.a) # Search libakg_ext.a in directory akg | |||
| if(NOT EXISTS ${AKG_EXTEND_FILE}) | |||
| set(AKG_EXTEND_FILE ${AKG_SOURCE_DIR}/build/libakg_ext.a) # Search libakg_ext.a in directory akg/build | |||
| if(NOT EXISTS ${AKG_EXTEND_FILE} AND NOT USE_KC_AIR) # Download libakg_ext.a to directory akg/build | |||
| set(AKG_EXTEND_FILE ) | |||
| set(LIB_PATH1 ${AKG_SOURCE_DIR}/libakg_ext.a) | |||
| set(LIB_PATH2 ${AKG_SOURCE_DIR}/build/libakg_ext.a) | |||
| if(EXISTS ${LIB_PATH1}) # Search libakg_ext.a in akg/ | |||
| set(AKG_EXTEND_FILE ${LIB_PATH1}) | |||
| else() | |||
| if(EXISTS ${LIB_PATH2}) # Search libakg_ext.a in akg/build/ | |||
| set(AKG_EXTEND_FILE ${LIB_PATH2}) | |||
| else() # Download libakg_ext.a to akg/build/ | |||
| execute_process(COMMAND bash ${AKG_SOURCE_DIR}/build.sh -a | |||
| WORKING_DIRECTORY ${AKG_SOURCE_DIR} | |||
| OUTPUT_VARIABLE OUTPUT_URL | |||
| RESULT_VARIABLE RESULT) | |||
| if(RESULT EQUAL 0) | |||
| set(AKG_EXTEND_FILE ${AKG_SOURCE_DIR}/build/libakg_ext.a) | |||
| if(RESULT EQUAL 0 AND OUTPUT_URL MATCHES "libakg_ext.a") | |||
| # Download library | |||
| string(STRIP ${OUTPUT_URL} LIB_URL) | |||
| message("-- Downloading ${LIB_URL} --> ${LIB_PATH2}") | |||
| file(DOWNLOAD ${LIB_URL} ${LIB_PATH2} STATUS DOWNLOAD_STATUS) | |||
| message("-- Download status: ${DOWNLOAD_STATUS}") | |||
| list(GET DOWNLOAD_STATUS 0 DOWNLOAD_CODE) | |||
| if(DOWNLOAD_CODE EQUAL 0) | |||
| set(AKG_EXTEND_FILE ${LIB_PATH2}) | |||
| endif() | |||
| endif() | |||
| endif() | |||
| endif() | |||
| message("-- AKG_EXTEND_FILE: ${AKG_EXTEND_FILE}") | |||
| if(EXISTS ${AKG_EXTEND_FILE}) | |||
| message("-- AKG_EXTEND_FILE: ${AKG_EXTEND_FILE}") | |||
| file(COPY ${AKG_EXTEND_FILE} DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/akg_extend) | |||
| execute_process(COMMAND ar -x libakg_ext.a | |||
| WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/akg_extend) | |||
| @@ -50,22 +50,23 @@ write_checksum() | |||
| done | |||
| } | |||
| download_lib() | |||
| acquire_lib_url() | |||
| { | |||
| uname_info=`uname -a | tr '[A-Z]' '[a-z]'` | |||
| os_info=`cat /etc/os-release | grep '^NAME=' | tr '[A-Z]' '[a-z]'` | |||
| os_name="" | |||
| arch_name="" | |||
| if [[ "${uname_info}" =~ "ubuntu" ]]; then | |||
| if [[ "${os_info}" =~ "ubuntu" ]]; then | |||
| os_name="ubuntu" | |||
| elif [[ "${uname_info}" =~ "euleros" ]]; then | |||
| elif [[ "${os_info}" =~ "euleros" ]]; then | |||
| os_name="euleros" | |||
| elif [[ "${uname_info}" =~ "centos" ]]; then | |||
| elif [[ "${os_info}" =~ "centos" ]]; then | |||
| os_name="centos" | |||
| fi | |||
| if [[ "${uname_info}" =~ "aarch64" ]]; then | |||
| arch_info=`arch | tr '[A-Z]' '[a-z]'` | |||
| arch_name="" | |||
| if [[ "${arch_info}" =~ "aarch64" ]]; then | |||
| arch_name="aarch64" | |||
| elif [[ "${uname_info}" =~ "x86_64" ]]; then | |||
| elif [[ "${arch_info}" =~ "x86" ]]; then | |||
| arch_name="x86" | |||
| fi | |||
| @@ -73,32 +74,7 @@ download_lib() | |||
| url_prefix="https://repo.mindspore.cn/public/ms-incubator/akg-binary/version" | |||
| lib_mark="202103/20210318/master_20210318142553_3e77f3a799ca87c23f1a906eaad5ec4c1f78bc95" | |||
| lib_url="${url_prefix}/${lib_mark}/lib/${os_arch}/libakg_ext.a" | |||
| hash_url="${url_prefix}/${lib_mark}/lib/${os_arch}/libakg_ext.a.sha256" | |||
| if [ ! -d ${BUILD_DIR} ]; then | |||
| mkdir -pv ${BUILD_DIR} | |||
| fi | |||
| # Download libakg_ext.a.sha256 | |||
| wget -P ${BUILD_DIR} --waitretry=10 --tries=3 ${hash_url} | |||
| if [ $? -ne 0 ]; then | |||
| echo "Fail to download ${hash_url}" | |||
| return 1 | |||
| fi | |||
| # Download libakg_ext.a | |||
| wget -P ${BUILD_DIR} --waitretry=10 --tries=3 ${lib_url} | |||
| if [ $? -ne 0 ]; then | |||
| echo "Fail to download ${lib_url}" | |||
| return 1 | |||
| fi | |||
| # Check hash | |||
| cur_hash=`sha256sum -b ${BUILD_DIR}/libakg_ext.a | awk '{print $1}'` | |||
| orig_hash=`grep libakg_ext.a ${BUILD_DIR}/libakg_ext.a.sha256 | awk '{print $1}'` | |||
| if [ "${cur_hash}" != "${orig_hash}" ]; then | |||
| echo "Hash check failed!" | |||
| return 1 | |||
| fi | |||
| echo "${lib_url}" | |||
| } | |||
| if [ ! -n "$1" ]; then | |||
| @@ -130,10 +106,7 @@ do | |||
| t) | |||
| ;; | |||
| a) | |||
| download_lib | |||
| if [ $? -ne 0 ]; then | |||
| exit 1 | |||
| fi | |||
| acquire_lib_url | |||
| exit 0 | |||
| ;; | |||
| *) | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"11288","input_desc":[[{"data_type":"float32","shape":[1024],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1024],"tensor_name":"input_0"}],[{"data_type":"float16","shape":[8192,1024],"tensor_name":"input_2"}]],"op":"Fused_Cast_Cast_Mul_TensorAdd___12292245117929986167","op_desc":[{"attr":[{"name":"dst_type","value":"float16"}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1024],"tensor_name":"input_0"}]],"name":"Cast","output_desc":[{"data_type":"float16","name":"output","shape":[1024],"tensor_name":"output_0_0"}]},{"attr":[{"name":"dst_type","value":"float16"}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1024],"tensor_name":"input_1"}]],"name":"Cast","output_desc":[{"data_type":"float16","name":"output","shape":[1024],"tensor_name":"output_0_1"}]},{"attr":[{"name":"x_shape","value":[8192,1024]},{"name":"y_shape","value":[1024]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float16","name":"x","shape":[8192,1024],"tensor_name":"input_2"}],[{"data_type":"float16","name":"y","shape":[1024],"tensor_name":"output_0_1"}]],"name":"Mul","output_desc":[{"data_type":"float16","name":"output","shape":[8192,1024],"tensor_name":"output_0_2"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float16","name":"x","shape":[8192,1024],"tensor_name":"output_0_2"}],[{"data_type":"float16","name":"y","shape":[1024],"tensor_name":"output_0_0"}]],"name":"TensorAdd","output_desc":[{"data_type":"float16","name":"output","shape":[8192,1024],"tensor_name":"output_0_3"}]}],"output_desc":[{"data_type":"float16","shape":[1024],"tensor_name":"output_0_1"},{"data_type":"float16","shape":[8192,1024],"tensor_name":"output_0_3"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"44349.44349","id":1550,"input_desc":[[{"data_type":"float16","format":"DefaultFormat","shape":[4,1024,1],"tensor_name":"input_2"}],[{"data_type":"float16","format":"DefaultFormat","shape":[1024],"tensor_name":"input_4"}],[{"data_type":"float16","format":"DefaultFormat","shape":[4,1024,1024],"tensor_name":"input_1"}],[{"data_type":"float32","format":"DefaultFormat","shape":[1024],"tensor_name":"input_0"}]],"op":"Fused_Cast_RealDiv_Mul_TensorAdd_split_16909220147165618805","op_desc":[{"attr":[{"data_type":"bool","name":"is_backed_cast","value":false},{"data_type":"str","name":"pri_format","value":"NC1HWC0"},{"data_type":"str","name":"dst_type","value":"float16"}],"impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1024],"tensor_name":"input_0"}]],"name":"Cast","output_desc":[{"data_type":"float16","format":"DefaultFormat","name":"output","shape":[1024],"tensor_name":"output_0_0"}]},{"attr":[{"data_type":"str","name":"pri_format","value":"DefaultFormat"}],"impl_path":"","input_desc":[[{"data_type":"float16","format":"DefaultFormat","name":"input_0","shape":[4,1024,1024],"tensor_name":"input_1"}],[{"data_type":"float16","format":"DefaultFormat","name":"input_1","shape":[4,1024,1],"tensor_name":"input_2"}]],"name":"RealDiv","output_desc":[{"data_type":"float16","format":"DefaultFormat","name":"output","shape":[4,1024,1024],"tensor_name":"output_0_1"}]},{"attr":[{"data_type":"str","name":"pri_format","value":"DefaultFormat"}],"impl_path":"","input_desc":[[{"data_type":"float16","format":"DefaultFormat","name":"input_0","shape":[4,1024,1024],"tensor_name":"output_0_1"}],[{"data_type":"float16","format":"DefaultFormat","name":"input_1","shape":[1024],"tensor_name":"input_4"}]],"name":"Mul","output_desc":[{"data_type":"float16","format":"DefaultFormat","name":"output","shape":[4,1024,1024],"tensor_name":"output_0_2"}]},{"attr":[{"data_type":"str","name":"pri_format","value":"DefaultFormat"}],"impl_path":"","input_desc":[[{"data_type":"float16","format":"DefaultFormat","name":"input_0","shape":[4,1024,1024],"tensor_name":"output_0_2"}],[{"data_type":"float16","format":"DefaultFormat","name":"input_1","shape":[1024],"tensor_name":"output_0_0"}]],"name":"TensorAdd","output_desc":[{"data_type":"float16","format":"DefaultFormat","name":"output","shape":[4,1024,1024],"tensor_name":"output_0_3"}]}],"output_desc":[{"data_type":"float16","format":"DefaultFormat","shape":[4,1024,1024],"tensor_name":"output_0_1"},{"data_type":"float16","format":"DefaultFormat","shape":[4,1024,1024],"tensor_name":"output_0_3"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"26625.40461","id":2555,"input_desc":[[{"data_type":"float32","format":"DefaultFormat","shape":[1,1],"tensor_name":"input_0"}],[{"data_type":"float32","format":"DefaultFormat","shape":[1024,1024],"tensor_name":"input_8"}]],"op":"Fused_ClipByNormNoDivSum_RealDiv_fusion_4181144419579591378","op_desc":[{"attr":[{"data_type":"str","name":"fusion","value":"SelectGT_000"}],"fusion":"SelectGT_000","impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1,1],"tensor_name":"input_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1],"tensor_name":"input_1","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1,1],"tensor_name":"output_0_0"}]},{"attr":[{"data_type":"str","name":"fusion","value":"SelectGT_000_end"}],"fusion":"SelectGT_000_end","impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1,1],"tensor_name":"output_0_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1,1],"tensor_name":"input_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_2","shape":[1],"tensor_name":"input_3","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1,1],"tensor_name":"output_0_1"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1,1],"tensor_name":"output_0_1"}]],"name":"Sqrt","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1,1],"tensor_name":"output_0_2"}]},{"attr":[{"data_type":"str","name":"fusion","value":"SelectGT_000_end"}],"fusion":"SelectGT_000_end","impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1,1],"tensor_name":"output_0_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1,1],"tensor_name":"output_0_2"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_2","shape":[1,1],"tensor_name":"input_0"}]],"name":"Select","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1,1],"tensor_name":"output_0_3"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1,1],"tensor_name":"output_0_3"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1],"tensor_name":"input_7","value":1.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1,1],"tensor_name":"output_0_4"}]},{"attr":[{"data_type":"str","name":"pri_format","value":"FRACTAL_NZ"}],"impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1024,1024],"tensor_name":"input_8"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1,1],"tensor_name":"output_0_4"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1024,1024],"tensor_name":"output_0_5"}]}],"output_desc":[{"data_type":"float32","format":"DefaultFormat","shape":[1024,1024],"tensor_name":"output_0_5"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"26625.40534","id":2566,"input_desc":[[{"data_type":"float32","format":"DefaultFormat","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","format":"DefaultFormat","shape":[4096],"tensor_name":"input_7"}]],"op":"Fused_ClipByNormNoDivSum_RealDiv_fusion_8238389606767005164","op_desc":[{"attr":[{"data_type":"str","name":"fusion","value":"SelectGT_000"}],"fusion":"SelectGT_000","impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1],"tensor_name":"input_1","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1],"tensor_name":"output_0_0"}]},{"attr":[{"data_type":"str","name":"fusion","value":"SelectGT_000_end"}],"fusion":"SelectGT_000_end","impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_2","shape":[1],"tensor_name":"input_3","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1],"tensor_name":"output_0_1"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1],"tensor_name":"output_0_1"}]],"name":"Sqrt","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1],"tensor_name":"output_0_2"}]},{"attr":[{"data_type":"str","name":"fusion","value":"SelectGT_000_end"}],"fusion":"SelectGT_000_end","impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_2","shape":[1],"tensor_name":"input_0"}]],"name":"Select","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1],"tensor_name":"output_0_3"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1],"tensor_name":"input_3","value":1.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[1],"tensor_name":"output_0_4"}]},{"attr":[{"data_type":"str","name":"pri_format","value":"NC1HWC0"}],"impl_path":"","input_desc":[[{"data_type":"float32","format":"DefaultFormat","name":"input_0","shape":[4096],"tensor_name":"input_7"}],[{"data_type":"float32","format":"DefaultFormat","name":"input_1","shape":[1],"tensor_name":"output_0_4"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","format":"DefaultFormat","name":"output","shape":[4096],"tensor_name":"output_0_5"}]}],"output_desc":[{"data_type":"float32","format":"DefaultFormat","shape":[4096],"tensor_name":"output_0_5"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"1907_sp_1847_1787_1728_1704_1680_1656_1587_sp_552_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[2],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[2],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_12073466097680829202","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0019","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0019_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0019","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0009_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[2]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[2],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[2],"tensor_name":"output_0_8"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[2],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[2],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[2],"tensor_name":"output_0_9"}],"process":"aicore"},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[2],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[2],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[2],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[2],"tensor_name":"output_0_10"}],"process":"aicore"}],"output_desc":[{"data_type":"float32","shape":[2],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"9739_sp_9282_8825_8368_7912_7889_7864_sp_2735_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[512,1024],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[512,1024],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_13492243466190004284","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}]},{"attr":null,"fusion":"SelectGT_0013","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}]},{"attr":null,"fusion":"SelectGT_0013_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}]},{"attr":null,"fusion":"SelectGT_0013","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}]},{"attr":null,"fusion":"SelectGT_0002_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}]},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}]},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[512,1024]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[512,1024],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[512,1024],"tensor_name":"output_0_8"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[512,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[512,1024],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[512,1024],"tensor_name":"output_0_9"}]},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[512,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[512,1024],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[512,1024],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[512,1024],"tensor_name":"output_0_10"}]}],"output_desc":[{"data_type":"float32","shape":[512,1024],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"1944_sp_1884_1824_1747_1716_1692_1668_1612_sp_655_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[4096],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[4096],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_1445905573061742177","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0016","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0016_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0016","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0006_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[4096]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[4096],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[4096],"tensor_name":"output_0_8"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[4096],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[4096],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[4096],"tensor_name":"output_0_9"}],"process":"aicore"},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[4096],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[4096],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[4096],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[4096],"tensor_name":"output_0_10"}],"process":"aicore"}],"output_desc":[{"data_type":"float32","shape":[4096],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"1957_sp_1897_1837_1732_1708_1684_1660_1593_sp_587_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[1024],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[1024],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_15689878575778426853","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0013","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0013_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0013","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0003_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[1024]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[1024],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1024],"tensor_name":"output_0_8"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[1024],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[1024],"tensor_name":"output_0_9"}],"process":"aicore"},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[1024],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[1024],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[1024],"tensor_name":"output_0_10"}],"process":"aicore"}],"output_desc":[{"data_type":"float32","shape":[1024],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"9298_sp_8841_8384_7927_7902_7879_7441_sp_2650_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[21128],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[21128],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_16040335705910473299","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}]},{"attr":null,"fusion":"SelectGT_0020","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}]},{"attr":null,"fusion":"SelectGT_0020_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}]},{"attr":null,"fusion":"SelectGT_0020","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}]},{"attr":null,"fusion":"SelectGT_0009_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}]},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}]},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[21128]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[21128],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[21128],"tensor_name":"output_0_8"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[21128],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[21128],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[21128],"tensor_name":"output_0_9"}]},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[21128],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[21128],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[21128],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[21128],"tensor_name":"output_0_10"}]}],"output_desc":[{"data_type":"float32","shape":[21128],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"1917_sp_1857_1797_1738_1712_1688_1664_1601_sp_621_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[30522],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[30522],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_3830386909471115343","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0018","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0018_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0018","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}],"process":"aicore"},{"attr":null,"fusion":"SelectGT_0008_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}],"process":"aicore"},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[30522]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[30522],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[30522],"tensor_name":"output_0_8"}],"process":"aicore"},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[30522],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[30522],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[30522],"tensor_name":"output_0_9"}],"process":"aicore"},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[30522],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[30522],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[30522],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[30522],"tensor_name":"output_0_10"}],"process":"aicore"}],"output_desc":[{"data_type":"float32","shape":[30522],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"9673_sp_9262_8851_8049_8024_8001_7586_sp_2782_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[1024,1024],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[1024,1024],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_4148121723898026533","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}]},{"attr":null,"fusion":"SelectGT_0014","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}]},{"attr":null,"fusion":"SelectGT_0014_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}]},{"attr":null,"fusion":"SelectGT_0014","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}]},{"attr":null,"fusion":"SelectGT_0004_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}]},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}]},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[1024,1024]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[1024,1024],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1024,1024],"tensor_name":"output_0_8"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1024,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[1024,1024],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[1024,1024],"tensor_name":"output_0_9"}]},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1024,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[1024,1024],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[1024,1024],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[1024,1024],"tensor_name":"output_0_10"}]}],"output_desc":[{"data_type":"float32","shape":[1024,1024],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"9290_sp_8833_8376_7919_7896_7873_7430_sp_2599_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[2,1024],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[2,1024],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_5080003035626701281","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}]},{"attr":null,"fusion":"SelectGT_0012","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}]},{"attr":null,"fusion":"SelectGT_0012_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}]},{"attr":null,"fusion":"SelectGT_0012","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}]},{"attr":null,"fusion":"SelectGT_0001_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}]},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}]},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[2,1024]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[2,1024],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[2,1024],"tensor_name":"output_0_8"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[2,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[2,1024],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[2,1024],"tensor_name":"output_0_9"}]},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[2,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[2,1024],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[2,1024],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[2,1024],"tensor_name":"output_0_10"}]}],"output_desc":[{"data_type":"float32","shape":[2,1024],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"9742_sp_9285_8828_8371_7914_7891_7868_sp_2752_construct","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_2"}],[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[],"tensor_name":"input_1"}],[{"data_type":"float32","shape":[1],"tensor_name":"input_14"}],[{"data_type":"float32","shape":[21128,1024],"tensor_name":"input_16"}],[{"data_type":"float32","shape":[21128,1024],"tensor_name":"input_17"}]],"op":"Fused_LambUpdateWithLR_8456945009561581117","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_1"}]],"name":"RealDiv","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}]},{"attr":null,"fusion":"SelectGT_0011","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_1"}]},{"attr":null,"fusion":"SelectGT_0011_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_6","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_2"}]},{"attr":null,"fusion":"SelectGT_0011","impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Greater","output_desc":[{"data_type":"bool","name":"output","shape":[],"tensor_name":"output_0_3"}]},{"attr":null,"fusion":"SelectGT_0000_end","impl_path":"","input_desc":[[{"data_type":"bool","name":"condition","shape":[1],"tensor_name":"output_0_3"}],[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_2"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_9","value":1.0}]],"name":"Select","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_4"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_4"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_11","value":10.0}]],"name":"Minimum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_5"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_5"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_3","value":0.0}]],"name":"Maximum","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_6"}]},{"attr":[{"name":"x_shape","value":[]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_6"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"input_14"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[1],"tensor_name":"output_0_7"}]},{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[21128,1024]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"output_0_7"}],[{"data_type":"float32","name":"y","shape":[21128,1024],"tensor_name":"input_16"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[21128,1024],"tensor_name":"output_0_8"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[21128,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[21128,1024],"tensor_name":"output_0_8"}]],"name":"Sub","output_desc":[{"data_type":"float32","name":"output","shape":[21128,1024],"tensor_name":"output_0_9"}]},{"attr":[{"name":"fake_output","value":true}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[21128,1024],"tensor_name":"input_17"}],[{"data_type":"float32","name":"y","shape":[21128,1024],"tensor_name":"output_0_9"}],[{"data_type":"float32","name":"z","shape":[21128,1024],"tensor_name":"output_0_9"}]],"name":"InplaceAssign","output_desc":[{"data_type":"float32","name":"output","shape":[21128,1024],"tensor_name":"output_0_10"}]}],"output_desc":[{"data_type":"float32","shape":[21128,1024],"tensor_name":"output_0_10"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"10313","input_desc":[[{"data_type":"float16","shape":[16,1,1,512],"tensor_name":"input_2"}],[{"data_type":"float16","shape":[16,16,512,512],"tensor_name":"input_1"}]],"op":"Fused_Mul_Mul_TensorAdd__4400644352246048056","op_desc":[{"attr":[{"name":"x_shape","value":[1]},{"name":"y_shape","value":[16,16,512,512]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float16","name":"x","shape":[1],"tensor_name":"input_0","value":0.125}],[{"data_type":"float16","name":"y","shape":[16,16,512,512],"tensor_name":"input_1"}]],"name":"Mul","output_desc":[{"data_type":"float16","name":"output","shape":[16,16,512,512],"tensor_name":"output_0_0"}]},{"attr":[{"name":"x_shape","value":[16,1,1,512]},{"name":"y_shape","value":[1]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float16","name":"x","shape":[16,1,1,512],"tensor_name":"input_2"}],[{"data_type":"float16","name":"y","shape":[1],"tensor_name":"input_3","value":-10000.0}]],"name":"Mul","output_desc":[{"data_type":"float16","name":"output","shape":[16,1,1,512],"tensor_name":"output_0_1"}]},{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float16","name":"x","shape":[16,1,1,512],"tensor_name":"output_0_1"}],[{"data_type":"float16","name":"y","shape":[16,16,512,512],"tensor_name":"output_0_0"}]],"name":"TensorAdd","output_desc":[{"data_type":"float16","name":"output","shape":[16,16,512,512],"tensor_name":"output_0_2"}]}],"output_desc":[{"data_type":"float16","shape":[16,1,1,512],"tensor_name":"output_0_1"},{"data_type":"float16","shape":[16,16,512,512],"tensor_name":"output_0_2"}],"platform":"AKG","process":"aicore"} | |||
| @@ -0,0 +1,707 @@ | |||
| { | |||
| "composite": true, | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "shape": [ | |||
| 32, | |||
| 16, | |||
| 56, | |||
| 56, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_14" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_0" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_3" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_9" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_12" | |||
| } | |||
| ] | |||
| ], | |||
| "op": "Fused_Poly_Schedule_Opt_001", | |||
| "op_desc": [ | |||
| { | |||
| "attr": [ | |||
| { | |||
| "name": "x_shape", | |||
| "value": [ | |||
| 256 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "y_shape", | |||
| "value": [ | |||
| 1 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "data_format", | |||
| "value": [ | |||
| "NC1HWC0", | |||
| "NC1HWC0" | |||
| ] | |||
| } | |||
| ], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_0" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1 | |||
| ], | |||
| "tensor_name": "input_1", | |||
| "value": 9.964923265215475e-06 | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_0" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": [ | |||
| { | |||
| "name": "x_shape", | |||
| "value": [ | |||
| 256 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "y_shape", | |||
| "value": [ | |||
| 256 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "data_format", | |||
| "value": [ | |||
| "NC1HWC0", | |||
| "NC1HWC0" | |||
| ] | |||
| } | |||
| ], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_0" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_0" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_1" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": [ | |||
| { | |||
| "name": "x_shape", | |||
| "value": [ | |||
| 256 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "y_shape", | |||
| "value": [ | |||
| 1 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "data_format", | |||
| "value": [ | |||
| "NC1HWC0", | |||
| "NC1HWC0" | |||
| ] | |||
| } | |||
| ], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_3" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1 | |||
| ], | |||
| "tensor_name": "input_1", | |||
| "value": 9.964923265215475e-06 | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_2" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_2" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_1" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Sub", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_3" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_3" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1 | |||
| ], | |||
| "tensor_name": "input_7", | |||
| "value": 9.999999747378752e-06 | |||
| } | |||
| ] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_4" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_4" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Sqrt", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_5" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_9" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_5" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "RealDiv", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_6" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": [ | |||
| { | |||
| "name": "x_shape", | |||
| "value": [ | |||
| 256 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "y_shape", | |||
| "value": [ | |||
| 256 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "data_format", | |||
| "value": [ | |||
| "NC1HWC0", | |||
| "NC1HWC0" | |||
| ] | |||
| } | |||
| ], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_6" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_0" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_7" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_12" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_7" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Sub", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_8" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": [ | |||
| { | |||
| "name": "x_shape", | |||
| "value": [ | |||
| 256 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "y_shape", | |||
| "value": [ | |||
| 32, | |||
| 256, | |||
| 56, | |||
| 56 | |||
| ] | |||
| }, | |||
| { | |||
| "name": "data_format", | |||
| "value": [ | |||
| "NC1HWC0", | |||
| "NC1HWC0" | |||
| ] | |||
| } | |||
| ], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_6" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 32, | |||
| 16, | |||
| 56, | |||
| 56, | |||
| 16 | |||
| ], | |||
| "tensor_name": "input_14" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 32, | |||
| 16, | |||
| 56, | |||
| 56, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_9" | |||
| } | |||
| ] | |||
| }, | |||
| { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [ | |||
| 32, | |||
| 16, | |||
| 56, | |||
| 56, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_9" | |||
| } | |||
| ], | |||
| [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [ | |||
| 1, | |||
| 16, | |||
| 1, | |||
| 1, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_8" | |||
| } | |||
| ] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [ | |||
| 32, | |||
| 16, | |||
| 56, | |||
| 56, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_10" | |||
| } | |||
| ] | |||
| } | |||
| ], | |||
| "output_desc": [ | |||
| { | |||
| "data_type": "float32", | |||
| "shape": [ | |||
| 32, | |||
| 16, | |||
| 56, | |||
| 56, | |||
| 16 | |||
| ], | |||
| "tensor_name": "output_0_10" | |||
| } | |||
| ], | |||
| "platform": "AKG", | |||
| "process": "aicore" | |||
| } | |||
| @@ -0,0 +1,320 @@ | |||
| { | |||
| "composite": true, | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float16", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "input_0" | |||
| }], | |||
| [{ | |||
| "data_type": "float16", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "input_9" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "input_7" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "input_2" | |||
| }] | |||
| ], | |||
| "op": "Fused_Poly_Schedule_Opt_002", | |||
| "op_desc": [{ | |||
| "attr": [{ | |||
| "name": "dst_type", | |||
| "value": "float32" | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float16", | |||
| "name": "x", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "input_0" | |||
| }] | |||
| ], | |||
| "name": "Cast", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_0" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "axis", | |||
| "value": [0, 2, 3] | |||
| }, { | |||
| "name": "keep_dims", | |||
| "value": true | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_0" | |||
| }] | |||
| ], | |||
| "name": "ReduceSum", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_1" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "input_2" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1], | |||
| "tensor_name": "input_3", | |||
| "value": 9.999999747378752e-05 | |||
| }] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_2" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_2" | |||
| }] | |||
| ], | |||
| "name": "Sqrt", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_3" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1], | |||
| "tensor_name": "input_5", | |||
| "value": 1.0 | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_3" | |||
| }] | |||
| ], | |||
| "name": "RealDiv", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_4" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [256] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "input_7" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1], | |||
| "tensor_name": "input_8", | |||
| "value": -1.0 | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_5" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "dst_type", | |||
| "value": "float32" | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float16", | |||
| "name": "x", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "input_9" | |||
| }] | |||
| ], | |||
| "name": "Cast", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_6" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_6" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_5" | |||
| }] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_7" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [32, 256, 56, 56] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [256] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_7" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_4" | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_8" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [32, 256, 56, 56] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [32, 256, 56, 56] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_0" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_8" | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_9" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "axis", | |||
| "value": [0, 2, 3] | |||
| }, { | |||
| "name": "keep_dims", | |||
| "value": true | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 16, 56, 56, 16], | |||
| "tensor_name": "output_0_9" | |||
| }] | |||
| ], | |||
| "name": "ReduceSum", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_10" | |||
| }] | |||
| }], | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_10" | |||
| }, { | |||
| "data_type": "float32", | |||
| "shape": [1, 16, 1, 1, 16], | |||
| "tensor_name": "output_0_1" | |||
| }], | |||
| "platform": "AKG", | |||
| "process": "aicore" | |||
| } | |||
| @@ -0,0 +1,500 @@ | |||
| { | |||
| "composite": true, | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float16", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "input_19" | |||
| }], | |||
| [{ | |||
| "data_type": "float16", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "input_14" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_7" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_5" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_3" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_9" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_0" | |||
| }] | |||
| ], | |||
| "op": "Fused_Poly_Schedule_Opt_003", | |||
| "op_desc": [{ | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_0" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1], | |||
| "tensor_name": "input_1", | |||
| "value": 9.999999747378752e-05 | |||
| }] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_0" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_0" | |||
| }] | |||
| ], | |||
| "name": "Sqrt", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_1" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_3" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_1" | |||
| }] | |||
| ], | |||
| "name": "RealDiv", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_2" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [64] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [1] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_5" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1], | |||
| "tensor_name": "input_6", | |||
| "value": -9.964923265215475e-06 | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_3" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [64] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [1] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_7" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1], | |||
| "tensor_name": "input_8", | |||
| "value": 9.964923265215475e-06 | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_4" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_9" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_1" | |||
| }] | |||
| ], | |||
| "name": "RealDiv", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_5" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [64] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [64] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_5" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_4" | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_6" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_6" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_3" | |||
| }] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_7" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "dst_type", | |||
| "value": "float32" | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float16", | |||
| "name": "x", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "input_14" | |||
| }] | |||
| ], | |||
| "name": "Cast", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_8" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [64] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [1] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "input_7" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1], | |||
| "tensor_name": "input_15", | |||
| "value": -9.964923265215475e-06 | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_9" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_9" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_1" | |||
| }] | |||
| ], | |||
| "name": "RealDiv", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_10" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [64] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [32, 64, 56, 56] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_10" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_8" | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_11" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "dst_type", | |||
| "value": "float32" | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float16", | |||
| "name": "x", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "input_19" | |||
| }] | |||
| ], | |||
| "name": "Cast", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_12" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_12" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_11" | |||
| }] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_13" | |||
| }] | |||
| }, { | |||
| "attr": null, | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_13" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_7" | |||
| }] | |||
| ], | |||
| "name": "TensorAdd", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_14" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "x_shape", | |||
| "value": [32, 64, 56, 56] | |||
| }, { | |||
| "name": "y_shape", | |||
| "value": [64] | |||
| }, { | |||
| "name": "data_format", | |||
| "value": ["NC1HWC0", "NC1HWC0"] | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_14" | |||
| }], | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "y", | |||
| "shape": [1, 4, 1, 1, 16], | |||
| "tensor_name": "output_0_2" | |||
| }] | |||
| ], | |||
| "name": "Mul", | |||
| "output_desc": [{ | |||
| "data_type": "float32", | |||
| "name": "output", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_15" | |||
| }] | |||
| }, { | |||
| "attr": [{ | |||
| "name": "dst_type", | |||
| "value": "float16" | |||
| }], | |||
| "impl_path": "", | |||
| "input_desc": [ | |||
| [{ | |||
| "data_type": "float32", | |||
| "name": "x", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_15" | |||
| }] | |||
| ], | |||
| "name": "Cast", | |||
| "output_desc": [{ | |||
| "data_type": "float16", | |||
| "name": "output", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_16" | |||
| }] | |||
| }], | |||
| "output_desc": [{ | |||
| "data_type": "float16", | |||
| "shape": [32, 4, 56, 56, 16], | |||
| "tensor_name": "output_0_16" | |||
| }], | |||
| "platform": "AKG", | |||
| "process": "aicore" | |||
| } | |||
| @@ -0,0 +1 @@ | |||
| {"composite":true,"composite_graph":"12299","input_desc":[[{"data_type":"float32","shape":[],"tensor_name":"input_0"}],[{"data_type":"float32","shape":[1216,30522],"tensor_name":"input_1"}]],"op":"Fused_Reciprocal_ReduceSum_Mul___1222261331617186059","op_desc":[{"attr":null,"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1],"tensor_name":"input_0"}]],"name":"Reciprocal","output_desc":[{"data_type":"float32","name":"output","shape":[],"tensor_name":"output_0_0"}]},{"attr":[{"name":"axis","value":[0]},{"name":"keep_dims","value":false}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[1216,30522],"tensor_name":"input_1"}]],"name":"ReduceSum","output_desc":[{"data_type":"float32","name":"output","shape":[30522],"tensor_name":"output_0_1"}]},{"attr":[{"name":"x_shape","value":[30522]},{"name":"y_shape","value":[]},{"name":"data_format","value":["DefaultFormat","DefaultFormat"]}],"impl_path":"","input_desc":[[{"data_type":"float32","name":"x","shape":[30522],"tensor_name":"output_0_1"}],[{"data_type":"float32","name":"y","shape":[1],"tensor_name":"output_0_0"}]],"name":"Mul","output_desc":[{"data_type":"float32","name":"output","shape":[30522],"tensor_name":"output_0_2"}]}],"output_desc":[{"data_type":"float32","shape":[1],"tensor_name":"output_0_0"},{"data_type":"float32","shape":[30522],"tensor_name":"output_0_2"}],"platform":"AKG","process":"aicore"} | |||