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