Lines Matching refs:hasTensor
98 inline bool hasTensor(IOperationExecutionContext* context, const uint32_t tensor) { in hasTensor() function
195 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
211 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
229 const bool cifgWeightsAllOrNone = (hasTensor(context, kInputToInputWeightsTensor) && in prepare()
230 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
231 (!hasTensor(context, kInputToInputWeightsTensor) && in prepare()
232 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
235 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
241 if (hasTensor(context, kCellToForgetWeightsTensor)) { in prepare()
247 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
254 const bool cifgUsed = !hasTensor(context, kInputToInputWeightsTensor); in prepare()
256 ((hasTensor(context, kCellToInputWeightsTensor) || cifgUsed) && in prepare()
257 hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
258 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
259 (!hasTensor(context, kCellToInputWeightsTensor) && in prepare()
260 !hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
261 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
265 NN_RET_CHECK(hasTensor(context, kInputGateBiasTensor)); in prepare()
270 NN_RET_CHECK(!hasTensor(context, kInputGateBiasTensor)) in prepare()
284 if (hasTensor(context, kProjectionWeightsTensor)) { in prepare()
291 if (hasTensor(context, kProjectionBiasTensor)) { in prepare()
306 if (hasTensor(context, kInputLayerNormTensor)) { in prepare()
312 if (hasTensor(context, kForgetLayerNormTensor)) { in prepare()
318 if (hasTensor(context, kCellLayerNormTensor)) { in prepare()
324 if (hasTensor(context, kOutputLayerNormTensor)) { in prepare()
331 NN_RET_CHECK(!hasTensor(context, kInputLayerNormTensor)) in prepare()
333 const bool layerNormWeightsAllOrNoneCifg = (hasTensor(context, kForgetLayerNormTensor) && in prepare()
334 hasTensor(context, kCellLayerNormTensor) && in prepare()
335 hasTensor(context, kOutputLayerNormTensor)) || in prepare()
336 (!hasTensor(context, kForgetLayerNormTensor) && in prepare()
337 !hasTensor(context, kCellLayerNormTensor) && in prepare()
338 !hasTensor(context, kOutputLayerNormTensor)); in prepare()
341 const bool layerNormWeightsAllOrNone = (hasTensor(context, kInputLayerNormTensor) && in prepare()
342 hasTensor(context, kForgetLayerNormTensor) && in prepare()
343 hasTensor(context, kCellLayerNormTensor) && in prepare()
344 hasTensor(context, kOutputLayerNormTensor)) || in prepare()
345 (!hasTensor(context, kInputLayerNormTensor) && in prepare()
346 !hasTensor(context, kForgetLayerNormTensor) && in prepare()
347 !hasTensor(context, kCellLayerNormTensor) && in prepare()
348 !hasTensor(context, kOutputLayerNormTensor)); in prepare()