Lines Matching refs:getInputShape

178     const Shape inputShape = context->getInputShape(kInputTensor);  in prepare()
185 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
190 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
196 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
202 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
206 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
212 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
218 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
222 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
236 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
242 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
248 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
266 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
274 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
277 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
280 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
285 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare()
292 const Shape projectionBiasShape = context->getInputShape(kProjectionBiasTensor); in prepare()
297 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare()
301 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
307 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in prepare()
313 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in prepare()
319 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in prepare()
325 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in prepare()
352 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in prepare()
356 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
367 const Shape inputShape = context->getInputShape(kInputTensor); in execute()
368 const Shape inputToInputWeightsShape = context->getInputShape(kInputToInputWeightsTensor); in execute()
370 context->getInputShape(kRecurrentToInputWeightsTensor); in execute()
371 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute()
372 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in execute()
373 const Shape inputToForgetWeightsShape = context->getInputShape(kInputToForgetWeightsTensor); in execute()
375 context->getInputShape(kRecurrentToForgetWeightsTensor); in execute()
376 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in execute()
377 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in execute()
378 const Shape inputToCellWeightsShape = context->getInputShape(kInputToCellWeightsTensor); in execute()
379 const Shape recurrentToCellWeightsShape = context->getInputShape(kRecurrentToCellWeightsTensor); in execute()
380 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in execute()
381 const Shape inputToOutputWeightsShape = context->getInputShape(kInputToOutputWeightsTensor); in execute()
383 context->getInputShape(kRecurrentToOutputWeightsTensor); in execute()
384 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute()
385 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in execute()
386 const Shape projectionWeightsShape = context->getInputShape(kProjectionWeightsTensor); in execute()
387 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in execute()
388 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in execute()