diff options
author | 2018-09-24 12:12:30 -0700 | |
---|---|---|
committer | 2018-09-24 12:17:12 -0700 | |
commit | 7919d64414ed47d217b8fc508d1be56b2a531a3c (patch) | |
tree | 9d6fd19ca3932d89743ab4643a56529afc974303 /BUILD | |
parent | f361fb8e4b4a9838e60a11ab45391c308bcb90da (diff) |
Wrap forward and backward pass in a defun for L2HMC.
Also a small bugfix to handle unknown shapes in backprop._num_elements.
Before:
entry {
name: "L2hmcBenchmark.eager_train_cpu_defun"
iters: 10
wall_time: 0.594115018845
extras {
key: "examples_per_sec"
value {
double_value: 336.635152548
}
}
}
After:
entry {
name: "L2hmcBenchmark.eager_train_cpu_defun"
iters: 10
wall_time: 0.322251081467
extras {
key: "examples_per_sec"
value {
double_value: 620.634069216
}
}
}
PiperOrigin-RevId: 214308142
Diffstat (limited to 'BUILD')
0 files changed, 0 insertions, 0 deletions