Searched defs:weightsDims (Results 1 – 2 of 2) sorted by relevance
203 const int32_t offset_column, const std::vector<uint32_t>& weightsDims, in assignWeightsSubmatrix()347 void QuantizedLSTMCell::concatenateWeights(const std::vector<uint32_t>& weightsDims, in concatenateWeights()
2933 uint32_t weightsDims[2] = {inputSize, numUnits}; in bidirectionlSequenceRNNTest() local3009 uint32_t weightsDims[2] = {inputSize, numUnits}; in unidirectionlSequenceRNNTest() local