#include <nn_ops.h>
Optional attribute setters for FusedBatchNormV2.
| 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. |
StringPiece tensorflow::ops::FusedBatchNormV2::Attrs::data_format_ = "NHWC"
float tensorflow::ops::FusedBatchNormV2::Attrs::epsilon_ = 0.0001f
bool tensorflow::ops::FusedBatchNormV2::Attrs::is_training_ = true
TF_MUST_USE_RESULT Attrs tensorflow::ops::FusedBatchNormV2::Attrs::DataFormat( StringPiece x )
The data format for x and y.
Either "NHWC" (default) or "NCHW".
Defaults to "NHWC"
TF_MUST_USE_RESULT Attrs tensorflow::ops::FusedBatchNormV2::Attrs::Epsilon( float x )
A small float number added to the variance of x.
Defaults to 0.0001
TF_MUST_USE_RESULT Attrs tensorflow::ops::FusedBatchNormV2::Attrs::IsTraining( bool x )
A bool value to indicate the operation is for training (default) or inference.
Defaults to true
© 2018 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-v2/attrs.html