Searched refs:zoneoutCell (Results 1 – 8 of 8) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/ |
H A D | device_manager_test.cpp | 1166 float zoneoutCell {0.0f}; variable 1170 hiddenSize, numLayers, numDirections, dropout, zoneoutCell, zoneoutHidden, projSize);
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/ |
H A D | inner_model_test.cpp | 2084 float zoneoutCell {0.0f}; variable 2088 hiddenSize, numLayers, numDirections, dropout, zoneoutCell, zoneoutHidden, projSize);
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/ |
H A D | inner_model_test.cpp | 2023 float zoneoutCell {0.0f}; variable 2027 hiddenSize, numLayers, numDirections, dropout, zoneoutCell, zoneoutHidden, projSize);
|
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | _neural_network_runtime.md | 650 …mDirections,LSTM网络的方向数。<br/>- dropout,除第一层外每层输入被丢弃的概率[0.0, 1.0]。<br/>- zoneoutCell,控制单元保持上次状态的概率,默… 874 | OH_NN_LSTM_ZONEOUT_CELL<sup>12+</sup> | 当张量作为LSTM算子的zoneoutCell参数时,使用本枚举值。 |
|
H A D | _neural_nework_runtime.md | 650 …mDirections,LSTM网络的方向数。<br/>- dropout,除第一层外每层输入被丢弃的概率[0.0, 1.0]。<br/>- zoneoutCell,控制单元保持上次状态的概率,默… 874 | OH_NN_LSTM_ZONEOUT_CELL<sup>12+</sup> | 当张量作为LSTM算子的zoneoutCell参数时,使用本枚举值。 |
|
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
H A D | lite_graph_to_hdi_model_v2_1.cpp | 816 lstm.zoneoutCell = mindspore::lite::MindIR_LSTM_GetZoneoutCell(primitive); in ConvertLSTM()
|
/ohos5.0/drivers/interface/nnrt/v2_1/ |
H A D | NodeAttrTypes.idl | 1333 float zoneoutCell;
|
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | _neural_network_runtime.md | 650 …f each layer except the first layer is discarded is [0.0, 1.0].<br>- **zoneoutCell**: probability … 874 | OH_NN_LSTM_ZONEOUT_CELL<sup>12+</sup> | Used when the tensor is used as the **zoneoutCell** param…
|