Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp93 inline bool hasTensor(IOperationExecutionContext* context, const uint32_t tensor) { in hasTensor() function
108 params.use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in getLSTMParams()
109 params.use_peephole = hasTensor(context, kCellToOutputWeightsTensor); in getLSTMParams()
110 params.use_layer_norm = hasTensor(context, kOutputLayerNormWeightsTensor); in getLSTMParams()
111 params.use_projection_weight = hasTensor(context, kProjectionWeightsTensor); in getLSTMParams()
112 params.use_projection_bias = hasTensor(context, kProjectionBiasTensor); in getLSTMParams()
215 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
231 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
249 const bool cifgWeightsAllOrNone = (hasTensor(context, kInputToInputWeightsTensor) && in prepare()
250 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
[all …]
DQLSTM.cpp98 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()
[all …]