Searched refs:hiddenStateDims (Results 1 – 1 of 1) sorted by relevance
3935 uint32_t hiddenStateDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest() local3963 .dimensions = hiddenStateDims, in bidirectionlSequenceRNNTest()4011 uint32_t hiddenStateDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest() local4036 .dimensions = hiddenStateDims, in unidirectionlSequenceRNNTest()