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/dnn.h | |
parent | 9477900946f923cb43ed76ed215490d01474bfe7 (diff) |
Merge changes from github.
Change: 152200430
Diffstat (limited to 'tensorflow/stream_executor/dnn.h')
-rw-r--r-- | tensorflow/stream_executor/dnn.h | 38 |
1 files changed, 34 insertions, 4 deletions
diff --git a/tensorflow/stream_executor/dnn.h b/tensorflow/stream_executor/dnn.h index d6b3f51705..c5805064f3 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -1283,19 +1283,47 @@ class DnnSupport { virtual bool DoPoolForward(Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, const dnn::BatchDescriptor& input_dimensions, + const DeviceMemory<double>& input_data, + const dnn::BatchDescriptor& output_dimensions, + DeviceMemory<double>* output_data) { + LOG(FATAL) << "DoPoolForward not implemented for double."; + return false; + } + + virtual bool DoPoolForward(Stream* stream, + const dnn::PoolingDescriptor& pooling_dimensions, + const dnn::BatchDescriptor& input_dimensions, const DeviceMemory<Eigen::half>& input_data, const dnn::BatchDescriptor& output_dimensions, - DeviceMemory<Eigen::half>* output_data) = 0; + DeviceMemory<Eigen::half>* output_data) { + LOG(FATAL) << "DoPoolForward not implemented for float16."; + return false; + } // Performs differentiation of the pooling operation. virtual bool DoPoolBackward(Stream* stream, 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) { + LOG(FATAL) << "DoPoolBackward not implemented."; + return false; + } + + virtual bool DoPoolBackward(Stream* stream, + 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, const DeviceMemory<float>& input_diff_data, - DeviceMemory<float>* output_diff_data) = 0; + DeviceMemory<float>* output_diff_data) { + LOG(FATAL) << "DoPoolBackward not implemented."; + return false; + } virtual bool DoPoolBackward(Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, @@ -1304,7 +1332,10 @@ class DnnSupport { const dnn::BatchDescriptor& output_dimensions, const DeviceMemory<Eigen::half>& output_data, const DeviceMemory<Eigen::half>& input_diff_data, - DeviceMemory<Eigen::half>* output_diff_data) = 0; + DeviceMemory<Eigen::half>* output_diff_data) { + LOG(FATAL) << "DoPoolBackward not implemented."; + return false; + } // Applies local response normalization to the values from // input_data and writes the result to output_data. See comments on @@ -1884,4 +1915,3 @@ class DnnSupport { } // namespace perftools #endif // TENSORFLOW_STREAM_EXECUTOR_DNN_H_ - |