Lines Matching refs:getInputShape
90 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
99 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
104 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
110 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
116 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
120 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
126 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
132 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
136 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
150 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
156 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
162 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
180 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
188 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
191 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
194 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
199 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare()
206 const Shape projectionBiasShape = context->getInputShape(kProjectionBiasTensor); in prepare()
211 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare()
215 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare()
221 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormWeightsTensor); in prepare()
227 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormWeightsTensor); in prepare()
233 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormWeightsTensor); in prepare()
239 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormWeightsTensor); in prepare()
268 Shape outputShape = context->getInputShape(kInputTensor); in prepare()
275 Shape outputStateOutTensor = context->getInputShape(kOutputStateInTensor); in prepare()
281 Shape cellStateOutTensor = context->getInputShape(kCellStateInTensor); in prepare()
292 const auto outputStateSize = getNumberOfElements(context->getInputShape(kOutputStateInTensor)); in execute()
293 const auto cellStateSize = getNumberOfElements(context->getInputShape(kCellStateInTensor)); in execute()
318 context->getInputShape(kInputTensor), in execute()
323 context->getInputShape(kInputToOutputWeightsTensor), in execute()
328 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
371 context->getInputShape(kInputTensor), in execute()
376 context->getInputShape(kInputToOutputWeightsTensor), in execute()
381 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()