Searched refs:stateShape (Results 1 – 3 of 3) sorted by relevance
57 Shape *stateShape, in Prepare() argument91 stateShape->type = inputShape.type; in Prepare()92 stateShape->dimensions = { batch_size, memory_size * num_filters }; in Prepare()93 stateShape->offset = inputShape.offset; in Prepare()94 stateShape->scale = inputShape.scale; in Prepare()
53 std::vector<RunTimeOperandInfo> &operands, Shape *stateShape,
1300 Shape stateShape, outputShape; in executeOperation() local1304 &stateShape, &outputShape) && in executeOperation()1305 setInfoAndAllocateIfNeeded(&stateOut, stateShape) && in executeOperation()