Searched refs:kInputLayerNormWeightsTensor (Results 1 – 4 of 4) sorted by relevance
220 if (hasTensor(context, kInputLayerNormWeightsTensor)) { in prepare()221 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormWeightsTensor); in prepare()245 NN_RET_CHECK(!hasTensor(context, kInputLayerNormWeightsTensor)) in prepare()257 (hasTensor(context, kInputLayerNormWeightsTensor) && in prepare()261 (!hasTensor(context, kInputLayerNormWeightsTensor) && in prepare()345 context->getInputBuffer<float>(kInputLayerNormWeightsTensor), in execute()398 context->getInputBuffer<_Float16>(kInputLayerNormWeightsTensor), in execute()
106 GetInput(operation, operands, kInputLayerNormWeightsTensor); // optional in LSTMCell()
69 constexpr uint32_t kInputLayerNormWeightsTensor = 24; // Optional variable
95 static constexpr int kInputLayerNormWeightsTensor = 23; variable