Searched refs:numDirections (Results 1 – 8 of 8) sorted by relevance
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | _neural_network_runtime.md | 650 …numDirections\*numLayers, 4\*hiddenSize, inputSize]。<br/>- wHh,隐藏层到隐藏层的权重参数,形状为[numDirections\*num… 872 | OH_NN_LSTM_NUM_DIRECTIONS<sup>12+</sup> | 当张量作为LSTM算子的numDirections参数时,使用本枚举值。 |
|
H A D | _neural_nework_runtime.md | 650 …numDirections\*numLayers, 4\*hiddenSize, inputSize]。<br/>- wHh,隐藏层到隐藏层的权重参数,形状为[numDirections\*num… 872 | OH_NN_LSTM_NUM_DIRECTIONS<sup>12+</sup> | 当张量作为LSTM算子的numDirections参数时,使用本枚举值。 |
|
/ohos5.0/drivers/interface/nnrt/v2_1/ |
H A D | NodeAttrTypes.idl | 1296 * shape is [numDirections * numLayers, 4 * hiddenSize, inputSize]. 1298 * shape is [numDirections * numLayers, 4 * hiddenSize, hiddenSize]. 1300 * shape is [numDirections * numLayers, 8 * hiddenSize]. 1301 …* * <b>hiddenState</b>, init state of hidden-layer, shape is [numDirections * numLayers, batchSize… 1302 …* * <b>cellState</b>, init state of cell, shape is [numDirections * numLayers, batchSize, hiddenSi… 1307 * shape is [seqLen, batchSize, numDirections * realHiddenSize]. 1309 * shape is [numDirections * numLayers, batchSize, realHiddenSize]. 1311 * shape is [numDirections * numLayers, batchSize, hiddenSize]. 1329 long numDirections;
|
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | _neural_network_runtime.md | 650 …numDirections\*numLayers, 4\*hiddenSize, inputSize].<br>- **wHh**: weight from the hidden layer to… 872 | OH_NN_LSTM_NUM_DIRECTIONS<sup>12+</sup> | Used when the tensor is used as the **numDirections** p… 873 | OH_NN_LSTM_DROPOUT<sup>12+</sup> | Used when the tensor is used as the **numDirections** paramete…
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/ |
H A D | device_manager_test.cpp | 1164 int64_t numDirections {0}; 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 | 2082 int64_t numDirections {0}; 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 | 2021 int64_t numDirections {0}; variable 2027 hiddenSize, numLayers, numDirections, dropout, zoneoutCell, zoneoutHidden, projSize);
|
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
H A D | lite_graph_to_hdi_model_v2_1.cpp | 814 lstm.numDirections = mindspore::lite::MindIR_LSTM_GetNumDirections(primitive); in ConvertLSTM()
|