Searched refs:checkpoint_format_ (Results 1 – 2 of 2) sorted by relevance
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/gxm/src/ |
H A D | Engine.cpp | 267 bn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 283 cn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 291 cn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 297 fcbn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 305 fcbn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 312 fn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 320 fn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 328 bn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 336 bn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() 342 fcbn->Checkpoint(tBuf, n, checkpoint_format_); in checkpoint() [all …]
|
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/gxm/include/ |
H A D | Engine.hpp | 91 string checkpoint_dir_, checkpoint_format_; member in MLEngine
|