Searched refs:kInputLayerNormWeightsTensor (Results 1 – 3 of 3) sorted by relevance
76 constexpr uint32_t kInputLayerNormWeightsTensor = 24; // Optional variable326 if (hasTensor(context, kInputLayerNormWeightsTensor)) { in prepare()327 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormWeightsTensor); in prepare()351 NN_RET_CHECK(!hasTensor(context, kInputLayerNormWeightsTensor)) in prepare()363 (hasTensor(context, kInputLayerNormWeightsTensor) && in prepare()367 (!hasTensor(context, kInputLayerNormWeightsTensor) && in prepare()451 context->getInputBuffer<float>(kInputLayerNormWeightsTensor), in execute()504 context->getInputBuffer<_Float16>(kInputLayerNormWeightsTensor), in execute()
96 static constexpr int kInputLayerNormWeightsTensor = 23; variable
106 GetInput(operation, operands, kInputLayerNormWeightsTensor); // optional in LSTMCell()