1111def checkpointDeriveAdamWFields =
1112 (lambda unrestricted storages : (family CheckpointParameterStorages) .
1113 (eliminate
1114 CheckpointParameterStorages
1115 (lambda unrestricted current : (family CheckpointParameterStorages) .
1116 (pi unrestricted cursor : (family ModelWord64) . (family CheckpointLearnerFieldsResult)))
1117 -- the scrutinee, which was missing: `eliminate F` takes motive,
1118 -- SCRUTINEE, then one branch per constructor. Without it the End
1119 -- branch's body sat in the scrutinee's slot and the eliminator was
1120 -- handed a `pi cursor : ModelWord64 . ...` where a family belongs.
1121 storages
1122 (branch
1123 CheckpointParameterStoragesEnd
1124 .
1125 (lambda unrestricted cursor : (family ModelWord64) .
1126 (constructor
1127 CheckpointLearnerFieldsResult
1128 CheckpointLearnerFieldsAccepted
1129 (constructor CheckpointFields CheckpointFieldsEnd)
1130 cursor
1131 zero)))
1132 (branch
1133 CheckpointParameterStoragesNext
1134 head
1135 tail
1136 ih_tail
1137 .
1138 (lambda unrestricted cursor : (family ModelWord64) .
1139 (eliminate
1140 CheckpointParameterStorage
1141 (lambda unrestricted current : (family CheckpointParameterStorage) .
1142 (family CheckpointLearnerFieldsResult))
1143 head
1144 (branch
1145 CheckpointParameterStorageValue
1146 role
1147 name
1148 parameterOffset
1149 parameterBytes
1150 elements
1151 domain
1152 initialization
1153 .
1154 (eliminate
1155 CheckpointParameterRole
1156 (lambda unrestricted current : (family CheckpointParameterRole) .
1157 (family CheckpointLearnerFieldsResult))
1158 role
1159 (branch
1160 CheckpointTrainableParameter
1161 .
1162 (eliminate
1163 CheckpointWord64Result
1164 (lambda unrestricted current : (family CheckpointWord64Result) .
1165 (family CheckpointLearnerFieldsResult))
1166 (checkpointAlignUp64 cursor)
1167 (branch
1168 CheckpointWord64Succeeded
1169 firstOffset
1170 .
1171 (eliminate
1172 CheckpointWord64Result
1173 (lambda unrestricted current : (family CheckpointWord64Result) .
1174 (family CheckpointLearnerFieldsResult))
1175 (checkpointMultiplyFourChecked elements)
1176 (branch
1177 CheckpointWord64Succeeded
1178 momentBytes
1179 .
1180 (eliminate
1181 CheckpointWord64Result
1182 (lambda unrestricted current : (family CheckpointWord64Result) .
1183 (family CheckpointLearnerFieldsResult))
1184 (checkpointAddChecked
1185 (constructor
1186 CheckpointErrorCode
1187 CheckpointLearnerMomentExtentOverflow)
1188 firstOffset
1189 momentBytes)
1190 (branch
1191 CheckpointWord64Succeeded
1192 afterFirst
1193 .
1194 (eliminate
1195 CheckpointWord64Result
1196 (lambda unrestricted current : (family CheckpointWord64Result) .
1197 (family CheckpointLearnerFieldsResult))
1198 (checkpointAlignUp64 afterFirst)
1199 (branch
1200 CheckpointWord64Succeeded
1201 secondOffset
1202 .
1203 (eliminate
1204 CheckpointWord64Result
1205 (lambda unrestricted current : (family CheckpointWord64Result) .
1206 (family CheckpointLearnerFieldsResult))
1207 (checkpointAddChecked
1208 (constructor
1209 CheckpointErrorCode
1210 CheckpointLearnerMomentExtentOverflow)
1211 secondOffset
1212 momentBytes)
1213 (branch
1214 CheckpointWord64Succeeded
1215 nextCursor
1216 .
1217 (eliminate
1218 CheckpointBytesResult
1219 (lambda unrestricted current : (family CheckpointBytesResult) .
1220 (family CheckpointLearnerFieldsResult))
1221 (checkpointMomentDomain elements)
1222 (branch
1223 CheckpointBytesSucceeded
1224 momentDomain
1225 .
1226 (eliminate
1227 CheckpointLearnerFieldsResult
1228 (lambda unrestricted current : (family CheckpointLearnerFieldsResult) .
1229 (family CheckpointLearnerFieldsResult))
1230 (ih_tail nextCursor)
1231 (branch
1232 CheckpointLearnerFieldsAccepted
1233 tailFields
1234 finalExtent
1235 tailCount
1236 .
1237 (constructor
1238 CheckpointLearnerFieldsResult
1239 CheckpointLearnerFieldsAccepted
1240 (constructor
1241 CheckpointFields
1242 CheckpointFieldsNext
1243 (checkpointMomentField
1244 (constructor CheckpointFieldKind CheckpointAdamWFirstMoment)
1245 b"first-moment"
1246 name
1247 firstOffset
1248 momentBytes
1249 elements
1250 momentDomain)
1251 (constructor
1252 CheckpointFields
1253 CheckpointFieldsNext
1254 (checkpointMomentField
1255 (constructor
1256 CheckpointFieldKind
1257 CheckpointAdamWSecondMoment)
1258 b"second-moment"
1259 name
1260 secondOffset
1261 momentBytes
1262 elements
1263 momentDomain)
1264 tailFields))
1265 finalExtent
1266 (succ (succ tailCount))))
1267 (branch
1268 CheckpointLearnerFieldsRejected
1269 code
1270 ordinal
1271 count
1272 .
1273 (constructor
1274 CheckpointLearnerFieldsResult
1275 CheckpointLearnerFieldsRejected
1276 code
1277 (succ ordinal)
1278 count))))
1279 (branch
1280 CheckpointBytesFailed
1281 code
1282 .
1283 (constructor
1284 CheckpointLearnerFieldsResult
1285 CheckpointLearnerFieldsRejected
1286 code
1287 zero
1288 zero))))
1289 (branch
1290 CheckpointWord64Failed
1291 code
1292 .
1293 (constructor
1294 CheckpointLearnerFieldsResult
1295 CheckpointLearnerFieldsRejected
1296 code
1297 zero
1298 zero))))
1299 (branch
1300 CheckpointWord64Failed
1301 code
1302 .
1303 (constructor
1304 CheckpointLearnerFieldsResult
1305 CheckpointLearnerFieldsRejected
1306 code
1307 zero
1308 zero))))
1309 (branch
1310 CheckpointWord64Failed
1311 code
1312 .
1313 (constructor
1314 CheckpointLearnerFieldsResult
1315 CheckpointLearnerFieldsRejected
1316 code
1317 zero
1318 zero))))
1319 (branch
1320 CheckpointWord64Failed
1321 code
1322 .
1323 (constructor
1324 CheckpointLearnerFieldsResult
1325 CheckpointLearnerFieldsRejected
1326 code
1327 zero
1328 zero))))
1329 (branch
1330 CheckpointWord64Failed
1331 code
1332 .
1333 (constructor
1334 CheckpointLearnerFieldsResult
1335 CheckpointLearnerFieldsRejected
1336 code
1337 zero
1338 zero))))
1339 (branch CheckpointFrozenParameter . (ih_tail cursor))
1340 (branch CheckpointPersistentState . (ih_tail cursor)))))))))The compiler supplied declaration spans and resolved links from this source snapshot. This page does not assert that this file belongs to a checked closure.