Searched refs:stateShape (Results 1 – 3 of 3) sorted by relevance
47 Shape *stateShape, in Prepare() argument82 stateShape->type = inputShape.type; in Prepare()83 stateShape->dimensions = { batch_size, memory_size * num_filters }; in Prepare()84 stateShape->offset = inputShape.offset; in Prepare()85 stateShape->scale = inputShape.scale; in Prepare()
42 Shape* stateShape, Shape* outputShape);
1161 Shape stateShape, outputShape; in executeOperation() local1164 success = SVDF::Prepare(operation, mOperands, &stateShape, &outputShape) && in executeOperation()1165 setInfoAndAllocateIfNeeded(&stateOut, stateShape, &result) && in executeOperation()