Searched refs:kInputLayerNormWeightsTensor (Results 1 – 3 of 3) sorted by relevance
74 constexpr uint32_t kInputLayerNormWeightsTensor = 24; // Optional variable313 if (hasTensor(context, kInputLayerNormWeightsTensor)) { in prepare()314 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormWeightsTensor); in prepare()338 NN_RET_CHECK(!hasTensor(context, kInputLayerNormWeightsTensor)) in prepare()350 (hasTensor(context, kInputLayerNormWeightsTensor) && in prepare()354 (!hasTensor(context, kInputLayerNormWeightsTensor) && in prepare()408 context->getInputBuffer<float>(kInputLayerNormWeightsTensor), in execute()450 context->getInputBuffer<_Float16>(kInputLayerNormWeightsTensor), in execute()
94 static constexpr int kInputLayerNormWeightsTensor = 23; variable
97 GetInput(operation, operands, kInputLayerNormWeightsTensor); // optional in LSTMCell()