Lines Matching refs:Shape
179 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
186 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
191 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
197 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
203 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
207 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
213 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
219 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
223 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
237 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
243 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
249 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
267 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
275 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
278 const Shape cellGateBiasShape = context->getInputShape(kCellGateBiasTensor); in prepare()
281 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
286 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare()
293 const Shape projectionBiasShape = context->getInputShape(kProjectionBiasTensor); in prepare()
298 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare()
302 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
308 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in prepare()
314 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in prepare()
320 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in prepare()
326 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in prepare()
353 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in prepare()
354 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
357 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()
358 Shape cellStateOutShape = context->getOutputShape(kCellStateOutTensor); in prepare()
369 const Shape inputShape = context->getInputShape(kInputTensor); in execute()
370 const Shape inputToInputWeightsShape = context->getInputShape(kInputToInputWeightsTensor); in execute()
371 const Shape recurrentToInputWeightsShape = in execute()
373 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute()
374 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in execute()
375 const Shape inputToForgetWeightsShape = context->getInputShape(kInputToForgetWeightsTensor); in execute()
376 const Shape recurrentToForgetWeightsShape = in execute()
378 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in execute()
379 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in execute()
380 const Shape inputToCellWeightsShape = context->getInputShape(kInputToCellWeightsTensor); in execute()
381 const Shape recurrentToCellWeightsShape = context->getInputShape(kRecurrentToCellWeightsTensor); in execute()
382 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in execute()
383 const Shape inputToOutputWeightsShape = context->getInputShape(kInputToOutputWeightsTensor); in execute()
384 const Shape recurrentToOutputWeightsShape = in execute()
386 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute()
387 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in execute()
388 const Shape projectionWeightsShape = context->getInputShape(kProjectionWeightsTensor); in execute()
389 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in execute()
390 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in execute()