()
| 473 | ) |
| 474 | |
| 475 | def add_dispatch_table(): |
| 476 | # This is not a good method, because the pickle module has been modified. |
| 477 | pickle.dispatch_table[core.eager.Tensor] = reduce_varbase |
| 478 | pickle.dispatch_table[EagerParamBase] = reduce_varbase |
| 479 | pickle.dispatch_table[core.DenseTensor] = reduce_DenseTensor |
| 480 | pickle.dispatch_table.update(dispatch_table_layer) |
| 481 | |
| 482 | def pop_dispatch_table(): |
| 483 | pickle.dispatch_table.pop(core.DenseTensor) |