diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-01-19 12:47:08 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-19 13:08:19 -0800 |
| commit | 598583c20dba57e4c99430d345b88dc63bc662f0 (patch) | |
| tree | 02940d56fee0e548f81f8861786cd9c45fe97507 /tensorflow/stream_executor/stream.h | |
| parent | 82542b6128c30aecd51d776f2be636993f99bd6a (diff) | |
Add convolve quantized ops to StreamExecutor API
Change: 144996696
Diffstat (limited to 'tensorflow/stream_executor/stream.h')
| -rw-r--r-- | tensorflow/stream_executor/stream.h | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/stream.h b/tensorflow/stream_executor/stream.h index 711eb3079a..8a8b4b1660 100644 --- a/tensorflow/stream_executor/stream.h +++ b/tensorflow/stream_executor/stream.h @@ -245,6 +245,26 @@ class Stream { const dnn::BatchDescriptor &output_descriptor, DeviceMemory<float> *output); + Stream &ThenConvolveQuantized( + const dnn::BatchDescriptor &input_descriptor, + const DeviceMemory<float> &input_data, + const dnn::FilterDescriptor &filter_descriptor, + const DeviceMemory<int8> &filter_coefficients, + const DeviceMemory<float> &coefficient_scales, + const dnn::ConvolutionDescriptor &convolution_descriptor, + const dnn::BatchDescriptor &output_descriptor, + DeviceMemory<float> *output_data); + + Stream &ThenConvolveQuantized( + const dnn::BatchDescriptor &input_descriptor, + const DeviceMemory<float> &input_data, + const dnn::FilterDescriptor &filter_descriptor, + const DeviceMemory<int16> &filter_coefficients, + const DeviceMemory<float> &coefficient_scales, + const dnn::ConvolutionDescriptor &convolution_descriptor, + const dnn::BatchDescriptor &output_descriptor, + DeviceMemory<float> *output_data); + Stream &ThenConvolveWithScratch( const dnn::BatchDescriptor &input_descriptor, const DeviceMemory<Eigen::half> &input_data, |