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