Lines Matching refs:numUnits

3928     const uint32_t numUnits = 5;  in bidirectionlSequenceRNNTest()  local
3931 uint32_t weightsDims[2] = {inputSize, numUnits}; in bidirectionlSequenceRNNTest()
3932 uint32_t recurrentWeightsDims[2] = {numUnits, numUnits}; in bidirectionlSequenceRNNTest()
3933 uint32_t biasDims[1] = {numUnits}; in bidirectionlSequenceRNNTest()
3934 uint32_t hiddenStateDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest()
3935 uint32_t outputDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest()
4004 const uint32_t numUnits = 5; in unidirectionlSequenceRNNTest() local
4007 uint32_t weightsDims[2] = {inputSize, numUnits}; in unidirectionlSequenceRNNTest()
4008 uint32_t recurrentWeightsDims[2] = {numUnits, numUnits}; in unidirectionlSequenceRNNTest()
4009 uint32_t biasDims[1] = {numUnits}; in unidirectionlSequenceRNNTest()
4010 uint32_t hiddenStateDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest()
4011 uint32_t outputDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest()
4068 const uint32_t numUnits = 4; in unidirectionalSequenceLSTMTest() local
4073 uint32_t inputWeightsDims[2] = {numUnits, inputSize}; in unidirectionalSequenceLSTMTest()
4074 uint32_t recurrentWeightsDims[2] = {numUnits, outputSize}; in unidirectionalSequenceLSTMTest()
4075 uint32_t diagonalDims[1] = {numUnits}; in unidirectionalSequenceLSTMTest()
4076 uint32_t projectionDims[2] = {outputSize, numUnits}; in unidirectionalSequenceLSTMTest()
4079 uint32_t cellStateDims[2] = {batchSize, numUnits}; in unidirectionalSequenceLSTMTest()