tensorflow::
#include <nn_ops.h>
Optional attribute setters for FusedBatchNorm.
Summary
Public attributes |
|
|---|---|
data_format_ = "NHWC"
|
StringPiece
|
epsilon_ = 0.0001f
|
float
|
is_training_ = true
|
bool
|
Public functions |
|
|---|---|
DataFormat(StringPiece x)
|
TF_MUST_USE_RESULT Attrs
The data format for x and y.
|
Epsilon(float x)
|
TF_MUST_USE_RESULT Attrs
A small float number added to the variance of x.
|
IsTraining(bool x)
|
TF_MUST_USE_RESULT Attrs
A bool value to indicate the operation is for training (default) or inference.
|
Public attributes
data_format_
StringPiece tensorflow::ops::FusedBatchNorm::Attrs::data_format_ = "NHWC"
epsilon_
float tensorflow::ops::FusedBatchNorm::Attrs::epsilon_ = 0.0001f
is_training_
bool tensorflow::ops::FusedBatchNorm::Attrs::is_training_ = true
Public functions
DataFormat
TF_MUST_USE_RESULT Attrs tensorflow::ops::FusedBatchNorm::Attrs::DataFormat( StringPiece x )
The data format for x and y.
Either "NHWC" (default) or "NCHW".
Defaults to "NHWC"
Epsilon
TF_MUST_USE_RESULT Attrs tensorflow::ops::FusedBatchNorm::Attrs::Epsilon( float x )
A small float number added to the variance of x.
Defaults to 0.0001
IsTraining
TF_MUST_USE_RESULT Attrs tensorflow::ops::FusedBatchNorm::Attrs::IsTraining( bool x )
A bool value to indicate the operation is for training (default) or inference.
Defaults to true