diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-01-24 10:05:12 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-24 10:27:38 -0800 |
| commit | eb65eff6cc65875c67e7dbf1c2e63baa6664bb6f (patch) | |
| tree | 7e92646fb31da8823f622e129e1199a31c89aead /tensorflow/stream_executor/dnn.cc | |
| parent | 7f79424f63c5684a43d47de216ae152144ddeecf (diff) | |
Add options argument for DNN activation
This is useful for platform-dependent functionality.
Change: 145432435
Diffstat (limited to 'tensorflow/stream_executor/dnn.cc')
| -rw-r--r-- | tensorflow/stream_executor/dnn.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index a1ce347497..e815a3bf74 100644 --- a/tensorflow/stream_executor/dnn.cc +++ b/tensorflow/stream_executor/dnn.cc @@ -64,6 +64,8 @@ string ActivationModeString(ActivationMode mode) { return "reluX"; case ActivationMode::kTanh: return "tanh"; + case ActivationMode::kBandPass: + return "bandpass"; default: LOG(FATAL) << "Unknown activation_mode " << static_cast<int32>(mode); } |