Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.cpp159 NN_CHECK_EQ(NumDimensions(input_to_input_weights), 2); in CheckInputTensorDimensions()
164 NN_CHECK_EQ(NumDimensions(input_to_forget_weights), 2); in CheckInputTensorDimensions()
168 NN_CHECK_EQ(NumDimensions(input_to_cell_weights), 2); in CheckInputTensorDimensions()
173 NN_CHECK_EQ(NumDimensions(recurrent_to_input_weights), 2); in CheckInputTensorDimensions()
178 NN_CHECK_EQ(NumDimensions(recurrent_to_forget_weights), 2); in CheckInputTensorDimensions()
182 NN_CHECK_EQ(NumDimensions(recurrent_to_cell_weights), 2); in CheckInputTensorDimensions()
194 NN_CHECK_EQ(NumDimensions(cell_to_input_weights), 1); in CheckInputTensorDimensions()
199 NN_CHECK_EQ(NumDimensions(cell_to_forget_weights), 1); in CheckInputTensorDimensions()
204 NN_CHECK_EQ(NumDimensions(cell_to_output_weights), 1); in CheckInputTensorDimensions()
231 NN_CHECK_EQ(NumDimensions(input_gate_bias), 1); in CheckInputTensorDimensions()
[all …]
DBidirectionalSequenceLSTM.cpp268 NN_CHECK(NumDimensions(input_) == 3); in Prepare()
274 NN_CHECK_EQ(NumDimensions(fw_input_to_output_weights_), 2); in Prepare()
277 NN_CHECK_EQ(NumDimensions(fw_recurrent_to_output_weights_), 2); in Prepare()
283 NN_CHECK_EQ(NumDimensions(bw_recurrent_to_output_weights_), 2); in Prepare()
346 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_input_weights_), 2); in Prepare()
350 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_forget_weights_), 2); in Prepare()
354 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_cell_weights_), 2); in Prepare()
358 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_output_weights_), 2); in Prepare()
365 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_input_weights_), 2); in Prepare()
369 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_forget_weights_), 2); in Prepare()
[all …]
DLSHProjection.cpp56 NN_CHECK_EQ(NumDimensions(hash), 2); in Prepare()
61 NN_CHECK(NumDimensions(input) >= 1); in Prepare()
75 NN_CHECK_EQ(NumDimensions(weight), 1); in Prepare()
DQuantizedLSTM.cpp251 NN_RET_CHECK_EQ(NumDimensions(input), 2); in prepare()
258 NN_RET_CHECK_EQ(NumDimensions(prevOutput), 2); in prepare()
270 NN_RET_CHECK_EQ(NumDimensions(weights), 2); in prepare()
302 NN_RET_CHECK_EQ(NumDimensions(bias), 1); in prepare()
318 NN_CHECK_EQ(NumDimensions(prevCellState), 2); in prepare()
/frameworks/ml/nn/common/include/
DCpuExecutor.h276 inline size_t NumDimensions(const RunTimeOperandInfo* operand) { in NumDimensions() function