Searched refs:kForgetLayerNormWeightsTensor (Results 1 – 3 of 3) sorted by relevance
77 constexpr uint32_t kForgetLayerNormWeightsTensor = 25; // Optional variable332 if (hasTensor(context, kForgetLayerNormWeightsTensor)) { in prepare()333 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormWeightsTensor); in prepare()354 (hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()357 (!hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()364 hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()368 !hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()452 context->getInputBuffer<float>(kForgetLayerNormWeightsTensor), in execute()505 context->getInputBuffer<_Float16>(kForgetLayerNormWeightsTensor), in execute()
97 static constexpr int kForgetLayerNormWeightsTensor = 24; variable
108 GetInput(operation, operands, kForgetLayerNormWeightsTensor); // optional in LSTMCell()