Lines Matching refs:hunits
396 int32_t hunits = ures_getInt(hunits_res.getAlias(), &status); in LSTMData() local
435 int32_t mat2_size = embedding_size * 4 * hunits; in LSTMData()
436 int32_t mat3_size = hunits * 4 * hunits; in LSTMData()
437 int32_t mat4_size = 4 * hunits; in LSTMData()
441 int32_t mat8_size = 2 * hunits * 4; in LSTMData()
450 fForwardW.init(data, embedding_size, 4 * hunits); in LSTMData()
452 fForwardU.init(data, hunits, 4 * hunits); in LSTMData()
454 fForwardB.init(data, 4 * hunits); in LSTMData()
456 fBackwardW.init(data, embedding_size, 4 * hunits); in LSTMData()
458 fBackwardU.init(data, hunits, 4 * hunits); in LSTMData()
460 fBackwardB.init(data, 4 * hunits); in LSTMData()
462 fOutputW.init(data, 2 * hunits, 4); in LSTMData()
609 int32_t hunits, in compute() argument
619 ifco.slice(0*hunits, hunits).sigmoid(); // i: sigmod in compute()
620 ifco.slice(1*hunits, hunits).sigmoid(); // f: sigmoid in compute()
621 ifco.slice(2*hunits, hunits).tanh(); // c_: tanh in compute()
622 ifco.slice(3*hunits, hunits).sigmoid(); // o: sigmod in compute()
624 c.hadamardProduct(ifco.slice(hunits, hunits)) in compute()
625 .addHadamardProduct(ifco.slice(0, hunits), ifco.slice(2*hunits, hunits)); in compute()
628 .hadamardProduct(ifco.slice(3*hunits, hunits)); in compute()
662 int32_t hunits = fData->fForwardU.d1(); in divideUpDictionaryRange() local
666 Array1D ifco(4 * hunits, status); in divideUpDictionaryRange()
668 Array1D c(hunits, status); in divideUpDictionaryRange()
674 Array2D hBackward(input_seq_len, hunits, status); in divideUpDictionaryRange()
677 Array1D fbRow(2 * hunits, status); in divideUpDictionaryRange()
698 compute(hunits, in divideUpDictionaryRange()
705 Array1D forwardRow = fbRow.slice(0, hunits); // point to first half of data in fbRow. in divideUpDictionaryRange()
706 Array1D backwardRow = fbRow.slice(hunits, hunits); // point to second half of data n fbRow. in divideUpDictionaryRange()
719 compute(hunits, in divideUpDictionaryRange()