Searched refs:dims (Results 1 – 3 of 3) sorted by relevance
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_utils.py | 62 def generate_tf_tensor(dims, minval, maxval, decimals=0, datatype=tf.float32): argument
63 array = minval + (maxval - minval) * np.random.rand(*dims)
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | lstm_settings.py | 345 dims = weight_tensor['shape']
346 row = dims[0]
347 col = dims[1]
|
| D | test_settings.py | 249 … def get_randomized_data(self, dims, npfile, regenerate, decimals=0, minrange=None, maxrange=None): argument
258 …data = tf.Variable(tf.random.uniform(dims, minval=minrange, maxval=maxrange, dtype=tf.dtypes.int64…
261 …data = tf.Variable(tf.random.uniform(dims, minval=minrange, maxval=maxrange, dtype=tf.dtypes.float…
|
