Searched refs:hiddenStateShape (Results 1 – 3 of 3) sorted by relevance
46 Shape *hiddenStateShape, in Prepare() argument75 hiddenStateShape->type = inputShape.type; in Prepare()76 hiddenStateShape->dimensions = { batch_size, num_units }; in Prepare()
34 Shape* hiddenStateShape, Shape* outputShape);
1150 Shape hiddenStateShape, outputShape; in executeOperation() local1153 success = RNN::Prepare(operation, mOperands, &hiddenStateShape, &outputShape) && in executeOperation()1154 setInfoAndAllocateIfNeeded(&hiddenStateOut, hiddenStateShape, &result) && in executeOperation()