Lines Matching refs:getInputShape
63 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
70 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
75 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
81 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
87 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
91 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
97 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
103 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
107 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
121 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
127 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
133 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
151 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
159 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
162 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
165 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
170 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare()
177 const Shape projectionBiasShape = context->getInputShape(kProjectionBiasTensor); in prepare()
182 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare()
186 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
192 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in prepare()
198 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in prepare()
204 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in prepare()
210 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in prepare()
237 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in prepare()
241 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
253 const Shape inputShape = context->getInputShape(kInputTensor); in execute()
254 const Shape inputToInputWeightsShape = context->getInputShape(kInputToInputWeightsTensor); in execute()
256 context->getInputShape(kRecurrentToInputWeightsTensor); in execute()
257 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute()
258 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in execute()
259 const Shape inputToForgetWeightsShape = context->getInputShape(kInputToForgetWeightsTensor); in execute()
261 context->getInputShape(kRecurrentToForgetWeightsTensor); in execute()
262 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in execute()
263 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in execute()
264 const Shape inputToCellWeightsShape = context->getInputShape(kInputToCellWeightsTensor); in execute()
265 const Shape recurrentToCellWeightsShape = context->getInputShape(kRecurrentToCellWeightsTensor); in execute()
266 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in execute()
267 const Shape inputToOutputWeightsShape = context->getInputShape(kInputToOutputWeightsTensor); in execute()
269 context->getInputShape(kRecurrentToOutputWeightsTensor); in execute()
270 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute()
271 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in execute()
272 const Shape projectionWeightsShape = context->getInputShape(kProjectionWeightsTensor); in execute()
273 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in execute()
274 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in execute()