Lines Matching refs:getInputShape

200     const Shape inputShape = context->getInputShape(kInputTensor);  in prepare()
208 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
213 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
219 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
225 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
229 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
235 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
241 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
245 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
259 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
265 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
271 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
289 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
297 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
300 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
303 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
308 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare()
315 const Shape projectionBiasShape = context->getInputShape(kProjectionBiasTensor); in prepare()
320 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare()
324 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare()
330 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormWeightsTensor); in prepare()
336 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormWeightsTensor); in prepare()
342 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormWeightsTensor); in prepare()
348 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormWeightsTensor); in prepare()
377 Shape outputShape = context->getInputShape(kInputTensor); in prepare()
384 Shape outputStateOutTensor = context->getInputShape(kOutputStateInTensor); in prepare()
390 Shape cellStateOutTensor = context->getInputShape(kCellStateInTensor); in prepare()
401 const auto outputStateSize = getNumberOfElements(context->getInputShape(kOutputStateInTensor)); in execute()
402 const auto cellStateSize = getNumberOfElements(context->getInputShape(kCellStateInTensor)); in execute()
427 context->getInputShape(kInputTensor), in execute()
432 context->getInputShape(kInputToOutputWeightsTensor), in execute()
437 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
480 context->getInputShape(kInputTensor), in execute()
485 context->getInputShape(kInputToOutputWeightsTensor), in execute()
490 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()