Searched refs:kCellLayerNormWeightsTensor (Results 1 – 4 of 4) sorted by relevance
232 if (hasTensor(context, kCellLayerNormWeightsTensor)) { in prepare()233 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormWeightsTensor); in prepare()249 hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()252 !hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()259 hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()263 !hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()347 context->getInputBuffer<float>(kCellLayerNormWeightsTensor), in execute()400 context->getInputBuffer<_Float16>(kCellLayerNormWeightsTensor), in execute()
110 GetInput(operation, operands, kCellLayerNormWeightsTensor); // optional in LSTMCell()
71 constexpr uint32_t kCellLayerNormWeightsTensor = 26; // Optional variable
97 static constexpr int kCellLayerNormWeightsTensor = 25; variable