def test_kaiming_uniform_initializer_static(self):
"""Test Kaiming unorm initializer for matrix multiply.
"""
self.static_test_kaiming_initializer_common(
init_inst=initializer.KaimingUniform(),
dtype="float32",
uniform=True,
is_conv=False)
def test_kaiming_uniform_initializer_conv_static(self):
"""Test Kaiming unorm initializer for convolutions.
"""
self.static_test_kaiming_initializer_common(
init_inst=initializer.KaimingUniform(),
dtype="float32",
uniform=True,
is_conv=True)
def test_kaiming_dygraph(self):
self.dygraph_test_kaiming_initializer_common(
init_inst=initializer.KaimingUniform(),
dtype="float32",
uniform=True)
self.dygraph_test_kaiming_initializer_common(
init_inst=initializer.KaimingNormal(),
dtype="float32",
uniform=False)