Searched refs:test_layer (Results 1 – 5 of 5) sorted by relevance
42 test_layer = multi_head_attention.MultiHeadAttention(50 output = test_layer(query=query, value=value)55 test_layer = multi_head_attention.MultiHeadAttention(59 output = test_layer(query, query)64 test_layer = multi_head_attention.MultiHeadAttention(68 output, coef = test_layer(query, query, return_attention_scores=True)74 test_layer = multi_head_attention.MultiHeadAttention(79 output, coef = test_layer(query, value, return_attention_scores=True)86 test_layer = multi_head_attention.MultiHeadAttention(93 output = test_layer(query=query, value=value, attention_mask=mask_tensor)[all …]
74 def test_layer(substrate, content_type): function107 asn1Object = test_layer(substrate, this_layer)112 asn1Object = test_layer(substrate, this_layer)116 asn1Object = test_layer(substrate, this_layer)
176 test_layer = layers.Dense(16, name='test_layer')177 a_test = test_layer(a)178 self.assertIs(test_layer.input, a)179 self.assertIs(test_layer.output, a_test)180 self.assertEqual(test_layer.input_shape, (None, 32))181 self.assertEqual(test_layer.output_shape, (None, 16))239 test_layer = PowersLayer()240 p1, p2 = test_layer(x) # pylint: disable=not-callable242 self.assertIs(test_layer.input, x)243 self._assertAllIs(test_layer.output, [p1, p2])[all …]
106 def test_layer(self): member in SplitUtilsTest
151 def test_layer(self, f32_layer_fn, input_shape, rtol=2e-3, atol=2e-3, member in LayerCorrectnessTest