Searched refs:input_rhs (Results 1 – 4 of 4) sorted by relevance
| /cmsis-nn-latest/Source/FullyConnectedFunctions/ |
| D | arm_batch_matmul_s8.c | 52 const int8_t *input_rhs, in arm_batch_matmul_s8() argument 87 input_rhs, in arm_batch_matmul_s8() 95 input_rhs, in arm_batch_matmul_s8() 118 input_rhs += inner_rhs_diff; in arm_batch_matmul_s8() 121 input_rhs += outer_rhs_diff; in arm_batch_matmul_s8()
|
| D | arm_batch_matmul_s16.c | 52 const int16_t *input_rhs, in arm_batch_matmul_s16() argument 81 input_rhs, in arm_batch_matmul_s16() 94 input_rhs += inner_rhs_diff; in arm_batch_matmul_s16() 97 input_rhs += outer_rhs_diff; in arm_batch_matmul_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_batch_matmul.py | 43 input_rhs = keras.layers.Input(batch_input_shape=input_shape_rhs) 45 … layer = 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/Include/ |
| D | arm_nnfunctions.h | 2837 const int8_t *input_rhs, 2874 const int16_t *input_rhs,
|