Lines Matching defs:elements
56 // Pointer to matrix elements, in row-major order
58 const void *elements;
83 matrix.elements = embeddings_weights(i);
120 matrix.elements = hidden_weights(i);
137 matrix.elements = hidden_bias_weights(i);
162 matrix.elements = softmax_weights(0);
179 matrix.elements = softmax_bias_weights(0);
232 // Returns pointer to elements of transpose(proto.embeddings(i)), in row-major
271 // Returns pointer to elements of proto.hidden_bias(i), in row-major order.
285 // Returns pointer to elements of transpose(proto.softmax()), in row-major
300 // Returns pointer to elements of proto.softmax_bias(), in row-major order.