Skip to content

Instantly share code, notes, and snippets.

@bjacob
Created April 22, 2022 19:28
Show Gist options
  • Save bjacob/4e2fd1a4bcedd29444a5091d40f327ff to your computer and use it in GitHub Desktop.
Save bjacob/4e2fd1a4bcedd29444a5091d40f327ff to your computer and use it in GitHub Desktop.
%1128 = "tosa.table"(%1127, %977) : (tensor<1x4x384x384xi16>, tensor<513xi16>) -> tensor<1x4x384x384xi32>
%1129 = "tosa.table"(%1127, %978) : (tensor<1x4x384x384xi16>, tensor<513xi16>) -> tensor<1x4x384x384xi32>
%1130 = "tosa.logical_left_shift"(%1128, %1007) : (tensor<1x4x384x384xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x384xi32>
%1131 = "tosa.arithmetic_right_shift"(%1129, %1008) {round = true} : (tensor<1x4x384x384xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x384xi32>
%1132 = "tosa.add"(%1131, %1009) : (tensor<1x4x384x384xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x384xi32>
%1133 = "tosa.add"(%1130, %1132) : (tensor<1x4x384x384xi32>, tensor<1x4x384x384xi32>) -> tensor<1x4x384x384xi32>
%1134 = "tosa.arithmetic_right_shift"(%1133, %1010) {round = true} : (tensor<1x4x384x384xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x384xi32>
%1135 = "tosa.reduce_sum"(%1134) {axis = 3 : i64} : (tensor<1x4x384x384xi32>) -> tensor<1x4x384x1xi32>
%1136 = "tosa.clz"(%1135) : (tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1137 = "tosa.sub"(%1136, %1011) : (tensor<1x4x384x1xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x1xi32>
%1138 = "tosa.logical_left_shift"(%1135, %1137) : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1139 = "tosa.mul"(%1138, %1012) {shift = 31 : i32} : (tensor<1x4x384x1xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x1xi32>
%1140 = "tosa.add"(%1139, %1013) : (tensor<1x4x384x1xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x1xi32>
%1141 = "tosa.mul"(%1140, %1138) {shift = 31 : i32} : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1142 = "tosa.sub"(%1014, %1141) : (tensor<1x1x1x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1143 = "tosa.mul"(%1140, %1142) {shift = 31 : i32} : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1144 = "tosa.mul"(%1143, %1015) {shift = 0 : i32} : (tensor<1x4x384x1xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x1xi32>
%1145 = "tosa.add"(%1140, %1144) : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1146 = "tosa.mul"(%1145, %1138) {shift = 31 : i32} : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1147 = "tosa.sub"(%1014, %1146) : (tensor<1x1x1x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1148 = "tosa.mul"(%1145, %1147) {shift = 31 : i32} : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1149 = "tosa.mul"(%1148, %1015) {shift = 0 : i32} : (tensor<1x4x384x1xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x1xi32>
%1150 = "tosa.add"(%1145, %1149) : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1151 = "tosa.mul"(%1150, %1138) {shift = 31 : i32} : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1152 = "tosa.sub"(%1014, %1151) : (tensor<1x1x1x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1153 = "tosa.mul"(%1150, %1152) {shift = 31 : i32} : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1154 = "tosa.mul"(%1153, %1015) {shift = 0 : i32} : (tensor<1x4x384x1xi32>, tensor<1x1x1x1xi32>) -> tensor<1x4x384x1xi32>
%1155 = "tosa.add"(%1150, %1154) : (tensor<1x4x384x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1156 = "tosa.mul"(%1133, %1155) {shift = 30 : i32} : (tensor<1x4x384x384xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x384xi32>
%1157 = "tosa.sub"(%1016, %1136) : (tensor<1x1x1x1xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x1xi32>
%1158 = "tosa.arithmetic_right_shift"(%1156, %1157) {round = true} : (tensor<1x4x384x384xi32>, tensor<1x4x384x1xi32>) -> tensor<1x4x384x384xi32>
%1159 = "tosa.rescale"(%1158) {double_round = false, input_zp = 0 : i32, multiplier = [1073741824 : i32], output_zp = -128 : i32, per_channel = false, scale32 = true, shift = [30 : i32]} : (tensor<1x4x384x384xi32>) -> tensor<1x4x384x384xi8>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment