Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp159 NN_CHECK_EQ(NumDimensions(input_to_input_weights), 2u); in CheckInputTensorDimensions()
164 NN_CHECK_EQ(NumDimensions(input_to_forget_weights), 2u); in CheckInputTensorDimensions()
168 NN_CHECK_EQ(NumDimensions(input_to_cell_weights), 2u); in CheckInputTensorDimensions()
173 NN_CHECK_EQ(NumDimensions(recurrent_to_input_weights), 2u); in CheckInputTensorDimensions()
178 NN_CHECK_EQ(NumDimensions(recurrent_to_forget_weights), 2u); in CheckInputTensorDimensions()
182 NN_CHECK_EQ(NumDimensions(recurrent_to_cell_weights), 2u); in CheckInputTensorDimensions()
194 NN_CHECK_EQ(NumDimensions(cell_to_input_weights), 1u); in CheckInputTensorDimensions()
199 NN_CHECK_EQ(NumDimensions(cell_to_forget_weights), 1u); in CheckInputTensorDimensions()
204 NN_CHECK_EQ(NumDimensions(cell_to_output_weights), 1u); in CheckInputTensorDimensions()
231 NN_CHECK_EQ(NumDimensions(input_gate_bias), 1u); in CheckInputTensorDimensions()
[all …]
DBidirectionalSequenceLSTM.cpp267 NN_CHECK(NumDimensions(input_) == 3); in Prepare()
273 NN_CHECK_EQ(NumDimensions(fw_input_to_output_weights_), 2u); in Prepare()
276 NN_CHECK_EQ(NumDimensions(fw_recurrent_to_output_weights_), 2u); in Prepare()
282 NN_CHECK_EQ(NumDimensions(bw_recurrent_to_output_weights_), 2u); in Prepare()
345 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_input_weights_), 2u); in Prepare()
349 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_forget_weights_), 2u); in Prepare()
353 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_cell_weights_), 2u); in Prepare()
357 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_output_weights_), 2u); in Prepare()
364 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_input_weights_), 2u); in Prepare()
368 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_forget_weights_), 2u); in Prepare()
[all …]
DLSHProjection.cpp55 NN_CHECK_EQ(NumDimensions(hash), 2u); in Prepare()
60 NN_CHECK(NumDimensions(input) >= 1); in Prepare()
74 NN_CHECK_EQ(NumDimensions(weight), 1u); in Prepare()
DQuantizedLSTM.cpp252 NN_RET_CHECK_EQ(NumDimensions(input), 2u); in prepare()
259 NN_RET_CHECK_EQ(NumDimensions(prevOutput), 2u); in prepare()
271 NN_RET_CHECK_EQ(NumDimensions(weights), 2u); in prepare()
303 NN_RET_CHECK_EQ(NumDimensions(bias), 1u); in prepare()
319 NN_CHECK_EQ(NumDimensions(prevCellState), 2u); in prepare()
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h275 inline size_t NumDimensions(const RunTimeOperandInfo* operand) { in NumDimensions() function