tensorflow::ops::FusedBatchNorm::Attrs

#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

© 2020 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/versions/r1.15/api_docs/cc/struct/tensorflow/ops/fused-batch-norm/attrs