Searched refs:outputValuesShape (Results 1 – 1 of 1) sorted by relevance
103 Shape outputValuesShape = inputShape; in prepare() local104 outputValuesShape.dimensions.back() = k; in prepare()109 return context->setOutputShape(kOutputValuesTensor, outputValuesShape) && in prepare()