diff options
Diffstat (limited to 'tensorflow/stream_executor/dnn.h')
-rw-r--r-- | tensorflow/stream_executor/dnn.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/dnn.h b/tensorflow/stream_executor/dnn.h index 3df5365c23..9eca5abe1a 100644 --- a/tensorflow/stream_executor/dnn.h +++ b/tensorflow/stream_executor/dnn.h @@ -469,6 +469,9 @@ enum class PadAlignment : int64 { // Returns a string representation of the given padding alignment. string PadAlignmentString(PadAlignment alignment); +// Print alignment to str. Needed to use CHECK_EQ between two PadAlignments. +std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment); + // Describes a convolution. // // Uses the named argument construction form: @@ -710,7 +713,7 @@ class PoolingDescriptor { class AlgorithmDesc { public: typedef int64 Index; - AlgorithmDesc() : algo_(kDefaultAlgorithm), tensor_ops_enabled_(false) {} + AlgorithmDesc() : algo_(kDefaultAlgorithm), tensor_ops_enabled_(true) {} AlgorithmDesc(Index a, bool use_tensor_ops) : algo_(a), tensor_ops_enabled_(use_tensor_ops) {} bool is_default() const { return algo_ == kDefaultAlgorithm; } |