tensorflow:: ops:: FusedBatchNorm:: Attrs

#include <nn_ops.h>

Optional attribute setters for FusedBatchNorm .

Summary

Public attributes

data_format_ = "NHWC"
StringPiece
epsilon_ = 0.0001f
float
exponential_avg_factor_ = 1.0f
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.
ExponentialAvgFactor (float x)
TF_MUST_USE_RESULT Attrs
Defaults to 1.
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

exponential_avg_factor_

float tensorflow::ops::FusedBatchNorm::Attrs::exponential_avg_factor_ = 1.0f

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

ExponentialAvgFactor

TF_MUST_USE_RESULT Attrs tensorflow::ops::FusedBatchNorm::Attrs::ExponentialAvgFactor(
  float x
)

Defaults to 1.

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