MCPcopy Create free account
hub / github.com/NVIDIA/FasterTransformer / layer_norm

Function layer_norm

examples/tensorflow/bert/utils/bert.py:33–35  ·  view source on GitHub ↗
(input_tensor, name=None)

Source from the content-addressed store, hash-verified

31
32
33def layer_norm(input_tensor, name=None):
34 return tf.contrib.layers.layer_norm(
35 inputs=input_tensor, begin_norm_axis=-1, begin_params_axis=-1, scope=name)
36
37
38def attention_layer(from_tensor,

Callers 1

tf_bertFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected