Home
last modified time | relevance | path

Searched refs:use_peephole (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/operations/
DLayerNormLSTMTest.cpp88 bool use_cifg, bool use_peephole, bool use_projection_weights, in LayerNormLSTMOpModel() argument
94 use_peephole_(use_peephole), in LayerNormLSTMOpModel()
DLSTMTest.cpp78 bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
83 use_cifg_(use_cifg), use_peephole_(use_peephole), in LSTMOpModel()
DLSTM.h35 bool use_peephole; member
DLSTM.cpp210 params->use_peephole = !IsNullInput(cell_to_output_weights); in CheckInputTensorDimensions()
845 if (params.use_peephole) { in LSTMStep()
864 if (params.use_peephole) { in LSTMStep()
908 if (params.use_peephole) { in LSTMStep()
DUnidirectionalSequenceLSTM.cpp102 params.use_peephole = hasTensor(context, kCellToOutputWeightsTensor); in getLSTMParams()