Searched refs:kForgetLayerNormWeightsTensor (Results 1 – 4 of 4) sorted by relevance
226 if (hasTensor(context, kForgetLayerNormWeightsTensor)) { in prepare()227 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormWeightsTensor); in prepare()248 (hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()251 (!hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()258 hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()262 !hasTensor(context, kForgetLayerNormWeightsTensor) && in prepare()346 context->getInputBuffer<float>(kForgetLayerNormWeightsTensor), in execute()399 context->getInputBuffer<_Float16>(kForgetLayerNormWeightsTensor), in execute()
108 GetInput(operation, operands, kForgetLayerNormWeightsTensor); // optional in LSTMCell()
70 constexpr uint32_t kForgetLayerNormWeightsTensor = 25; // Optional variable
96 static constexpr int kForgetLayerNormWeightsTensor = 24; variable