Searched refs:cols_weights (Results 1 – 1 of 1) sorted by relevance
| /external/ComputeLibrary/tests/validation/reference/ |
| D | FullyConnectedLayer.cpp | 45 …nst SimpleTensor<TB> &bias, SimpleTensor<T> &dst, int offset_src, int offset_dst, int cols_weights, in vector_matrix_multiply() argument
57 …dst_ptr[y] = std::inner_product(src_ptr, src_ptr + cols_weights, &weights_ptr[cols_weights * y], s… in vector_matrix_multiply()
64 int cols_weights, int rows_weights) in vector_matrix_multiply() argument
97 for(int x = 0; x < cols_weights; ++x) in vector_matrix_multiply()
99 … acc += (src_ptr[x] + input_offset) * (weights_ptr[x + y * cols_weights] + weights_offset); in vector_matrix_multiply()
139 const int cols_weights = weights.shape().x(); in fully_connected_layer() local
145 const int offset_in = k * cols_weights; in fully_connected_layer()
154 cols_weights, in fully_connected_layer()
|