diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-04-04 16:10:08 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-04 17:24:57 -0700 |
| commit | ccbc8991db3943ef984405881a1c917c530f902f (patch) | |
| tree | a7b5c760155bfa4ff95ffc0ebd3823c649668997 /tensorflow/stream_executor/stream.h | |
| parent | 9477900946f923cb43ed76ed215490d01474bfe7 (diff) | |
Merge changes from github.
Change: 152200430
Diffstat (limited to 'tensorflow/stream_executor/stream.h')
| -rw-r--r-- | tensorflow/stream_executor/stream.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/stream.h b/tensorflow/stream_executor/stream.h index f22fba1d74..5b46b86f54 100644 --- a/tensorflow/stream_executor/stream.h +++ b/tensorflow/stream_executor/stream.h @@ -467,6 +467,12 @@ class Stream { Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions, const dnn::BatchDescriptor &input_dimensions, + const DeviceMemory<double> &input_data, + const dnn::BatchDescriptor &output_dimensions, + DeviceMemory<double> *output_data); + + Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions, + const dnn::BatchDescriptor &input_dimensions, const DeviceMemory<float> &input_data, const dnn::BatchDescriptor &output_dimensions, DeviceMemory<float> *output_data); @@ -479,6 +485,14 @@ class Stream { Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions, const dnn::BatchDescriptor &input_dimensions, + const DeviceMemory<double> &input_data, + const dnn::BatchDescriptor &output_dimensions, + const DeviceMemory<double> &output_data, + const DeviceMemory<double> &input_diff_data, + DeviceMemory<double> *output_diff_data); + + Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions, + const dnn::BatchDescriptor &input_dimensions, const DeviceMemory<float> &input_data, const dnn::BatchDescriptor &output_dimensions, const DeviceMemory<float> &output_data, |