diff options
author | 2018-08-02 12:41:52 +0800 | |
---|---|---|
committer | 2018-08-02 12:41:52 +0800 | |
commit | 8c23e1ad36783085eff4ea0a33f66f0c6700f8af (patch) | |
tree | e0f50326b96953ff025d00b7f27f3bdef23633da /tensorflow/core/util/tensor_format.cc | |
parent | ad72f89ca9a92fdae4ae5147ed5bbe42b49adce0 (diff) |
Conv3d enhancement
Diffstat (limited to 'tensorflow/core/util/tensor_format.cc')
-rw-r--r-- | tensorflow/core/util/tensor_format.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/util/tensor_format.cc b/tensorflow/core/util/tensor_format.cc index a5f7ecf0d1..f331973f5c 100644 --- a/tensorflow/core/util/tensor_format.cc +++ b/tensorflow/core/util/tensor_format.cc @@ -25,6 +25,10 @@ string GetConvnet3dDataFormatAttrString() { return "data_format: { 'NDHWC', 'NCDHW' } = 'NDHWC' "; } +string GetConvnetDataFormat2D3DAttrString() { + return "data_format: { 'NHWC', 'NCHW', 'NDHWC', 'NCDHW' } = 'NHWC' "; +} + string GetConvnetFilterFormatAttrString() { return "filter_format: { 'HWIO', 'OIHW' } = 'HWIO' "; } |