#include <nn_ops.h>
Optional attribute setters for FusedBatchNorm.
Public attributes | |
---|---|
data_format_ = "NHWC" | StringPiece |
epsilon_ = 0.0001f | float |
is_training_ = true | bool |
Public functions | |
---|---|
DataFormat(StringPiece x) | The data format for x and y. |
Epsilon(float x) | A small float number added to the variance of x. |
IsTraining(bool x) | A bool value to indicate the operation is for training (default) or inference. |
StringPiece tensorflow::ops::FusedBatchNorm::Attrs::data_format_ = "NHWC"
float tensorflow::ops::FusedBatchNorm::Attrs::epsilon_ = 0.0001f
bool tensorflow::ops::FusedBatchNorm::Attrs::is_training_ = true
Attrs tensorflow::ops::FusedBatchNorm::Attrs::DataFormat( StringPiece x )
The data format for x and y.
Either "NHWC" (default) or "NCHW".
Defaults to "NHWC"
Attrs tensorflow::ops::FusedBatchNorm::Attrs::Epsilon( float x )
A small float number added to the variance of x.
Defaults to 0.0001
Attrs tensorflow::ops::FusedBatchNorm::Attrs::IsTraining( bool x )
A bool value to indicate the operation is for training (default) or inference.
Defaults to true
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/struct/tensorflow/ops/fused-batch-norm/attrs.html