Home
last modified time | relevance | path

Searched refs:layer (Results 1 – 5 of 5) sorted by relevance

/cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/
Dop_transpose.py41 layer = tf.transpose(input_lhs, perm=params["perm"])
42 model = keras.Model([input_lhs], [layer])
Dop_maximum_minimum.py51 layer = layer_type(input_1, input_2)
52 model = keras.Model([input_1, input_2], [layer])
Dop_batch_matmul.py45layer = tf.matmul(input_lhs, input_rhs, transpose_a=params["adj_x"], transpose_b=params["adj_y"])
46 model = keras.Model([input_lhs, input_rhs], [layer])
/cmsis-nn-latest/Tests/UnitTest/
Dadd_mul_settings.py97 layer = keras.layers.Add()([input1, input2])
99 layer = keras.layers.Multiply()([input1, input2])
102 out = keras.layers.Lambda(function=lambda x: x)(layer)
Dsvdf_settings.py255 def get_scale_and_zp(self, layer): argument
256 …return (layer['quantization_parameters']['scales'][0], layer['quantization_parameters']['zero_poin…