Home
last modified time | relevance | path

Searched refs:hasTensor (Results 1 – 2 of 2) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp218 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
234 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
258 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
264 if (hasTensor(context, kCellToForgetWeightsTensor)) { in prepare()
270 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
307 if (hasTensor(context, kProjectionWeightsTensor)) { in prepare()
314 if (hasTensor(context, kProjectionBiasTensor)) { in prepare()
329 if (hasTensor(context, kInputLayerNormWeightsTensor)) { in prepare()
335 if (hasTensor(context, kForgetLayerNormWeightsTensor)) { in prepare()
341 if (hasTensor(context, kCellLayerNormWeightsTensor)) { in prepare()
[all …]
H A DQLSTM.cpp196 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
212 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
236 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
242 if (hasTensor(context, kCellToForgetWeightsTensor)) { in prepare()
248 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
285 if (hasTensor(context, kProjectionWeightsTensor)) { in prepare()
292 if (hasTensor(context, kProjectionBiasTensor)) { in prepare()
307 if (hasTensor(context, kInputLayerNormTensor)) { in prepare()
313 if (hasTensor(context, kForgetLayerNormTensor)) { in prepare()
319 if (hasTensor(context, kCellLayerNormTensor)) { in prepare()
[all …]