Searched defs:cellLayerNormShape (Results 1 – 2 of 2) sorted by relevance
204 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in prepare() local266 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in execute() local
233 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormWeightsTensor); in prepare() local