tensorflow:: ops:: Conv:: Attrs
#include <nn_ops.h>
Optional attribute setters for Conv.
Summary
Public attributes |
|
---|---|
batch_dims_ = 1
|
int64
|
data_format_ = "CHANNELS_LAST"
|
StringPiece
|
dilations_ = {}
|
gtl::ArraySlice< int >
|
explicit_paddings_ = {}
|
gtl::ArraySlice< int >
|
groups_ = 1
|
int64
|
Public functions |
|
---|---|
BatchDims(int64 x)
|
TF_MUST_USE_RESULT Attrs
A positive integer specifying the number of batch dimensions for the input tensor.
|
DataFormat(StringPiece x)
|
TF_MUST_USE_RESULT Attrs
Used to set the data format.
|
Dilations(const gtl::ArraySlice< int > & x)
|
TF_MUST_USE_RESULT Attrs
1-D tensor of length
N+2 . |
ExplicitPaddings(const gtl::ArraySlice< int > & x)
|
TF_MUST_USE_RESULT Attrs
If
padding is "EXPLICIT" , the list of explicit padding amounts. |
Groups(int64 x)
|
TF_MUST_USE_RESULT Attrs
A positive integer specifying the number of groups in which the input is split along the channel axis.
|
Public attributes
batch_dims_
int64 tensorflow::ops::Conv::Attrs::batch_dims_ = 1
data_format_
StringPiece tensorflow::ops::Conv::Attrs::data_format_ = "CHANNELS_LAST"
dilations_
gtl::ArraySlice< int > tensorflow::ops::Conv::Attrs::dilations_ = {}
explicit_paddings_
gtl::ArraySlice< int > tensorflow::ops::Conv::Attrs::explicit_paddings_ = {}
groups_
int64 tensorflow::ops::Conv::Attrs::groups_ = 1
Public functions
BatchDims
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv::Attrs::BatchDims( int64 x )
A positive integer specifying the number of batch dimensions for the input tensor.
Should be less than the rank of the input tensor.
Defaults to 1
DataFormat
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv::Attrs::DataFormat( StringPiece x )
Used to set the data format.
By default CHANNELS_FIRST
, uses NHWC (2D) / NDHWC (3D)
or if CHANNELS_LAST
, uses NCHW (2D) / NCDHW (3D)
.
Defaults to "CHANNELS_LAST"
Dilations
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv::Attrs::Dilations( const gtl::ArraySlice< int > & x )
1-D tensor of length N+2
.
The dilation factor for each dimension of input
. If set to k > 1
, there will be k-1
skipped cells between each filter element on that dimension. The dimension order is determined by the value of channels_last_format
, see above for details. Dilations in the batch and depth dimensions must be 1.
Defaults to []
ExplicitPaddings
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv::Attrs::ExplicitPaddings( const gtl::ArraySlice< int > & x )
If padding
is "EXPLICIT"
, the list of explicit padding amounts.
For the ith dimension, the amount of padding inserted before and after the dimension is explicit_paddings[2 * i]
and explicit_paddings[2 * i + 1]
, respectively. If padding
is not "EXPLICIT"
, explicit_paddings
must be empty.
Defaults to []
Groups
TF_MUST_USE_RESULT Attrs tensorflow::ops::Conv::Attrs::Groups( int64 x )
A positive integer specifying the number of groups in which the input is split along the channel axis.
Each group is convolved separately with filters / groups
filters. The output is the concatenation of all the groups results along the channel axis. Input channels and filters must both be divisible by groups.
Defaults to 1