Home
last modified time | relevance | path

Searched refs:NN_CHECK_EQ (Results 1 – 8 of 8) sorted by relevance

/packages/modules/NeuralNetworks/common/operations/
DLSTM.cpp159 NN_CHECK_EQ(NumDimensions(input_to_input_weights), 2); in CheckInputTensorDimensions()
160 NN_CHECK_EQ(SizeOfDimension(input_to_input_weights, 0), n_cell); in CheckInputTensorDimensions()
161 NN_CHECK_EQ(SizeOfDimension(input_to_input_weights, 1), n_input); in CheckInputTensorDimensions()
164 NN_CHECK_EQ(NumDimensions(input_to_forget_weights), 2); in CheckInputTensorDimensions()
165 NN_CHECK_EQ(SizeOfDimension(input_to_forget_weights, 0), n_cell); in CheckInputTensorDimensions()
166 NN_CHECK_EQ(SizeOfDimension(input_to_forget_weights, 1), n_input); in CheckInputTensorDimensions()
168 NN_CHECK_EQ(NumDimensions(input_to_cell_weights), 2); in CheckInputTensorDimensions()
169 NN_CHECK_EQ(SizeOfDimension(input_to_cell_weights, 0), n_cell); in CheckInputTensorDimensions()
170 NN_CHECK_EQ(SizeOfDimension(input_to_cell_weights, 1), n_input); in CheckInputTensorDimensions()
173 NN_CHECK_EQ(NumDimensions(recurrent_to_input_weights), 2); in CheckInputTensorDimensions()
[all …]
DLSHProjection.cpp52 NN_CHECK_EQ(NumOutputs(operation), 1); in Prepare()
55 NN_CHECK_EQ(NumDimensions(hash), 2); in Prepare()
73 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 4); in Prepare()
74 NN_CHECK_EQ(NumDimensions(weight), 1); in Prepare()
75 NN_CHECK_EQ(SizeOfDimension(weight, 0), SizeOfDimension(input, 0)); in Prepare()
DSVDF.cpp64 NN_CHECK_EQ(NumOutputs(operation), 2); in Prepare()
83 NN_CHECK_EQ(num_filters % rank, 0); in Prepare()
86 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(weights_feature, 1)); in Prepare()
87 NN_CHECK_EQ(SizeOfDimension(weights_time, 0), num_filters); in Prepare()
91 NN_CHECK_EQ(SizeOfDimension(bias, 0), num_units); in Prepare()
DRNN.cpp51 NN_CHECK_EQ(NumOutputs(operation), 2); in Prepare()
63 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(input_weights, 1)); in Prepare()
64 NN_CHECK_EQ(SizeOfDimension(input_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
65 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
66 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 1), SizeOfDimension(bias, 0)); in Prepare()
DBidirectionalSequenceLSTM.cpp273 NN_CHECK_EQ(NumDimensions(fw_input_to_output_weights_), 2); in Prepare()
274 NN_CHECK_EQ(SizeOfDimension(fw_input_to_output_weights_, 1), n_fw_input); in Prepare()
276 NN_CHECK_EQ(NumDimensions(fw_recurrent_to_output_weights_), 2); in Prepare()
277 NN_CHECK_EQ(SizeOfDimension(fw_recurrent_to_output_weights_, 0), n_fw_cell); in Prepare()
282 NN_CHECK_EQ(NumDimensions(bw_recurrent_to_output_weights_), 2); in Prepare()
283 NN_CHECK_EQ(SizeOfDimension(bw_recurrent_to_output_weights_, 0), n_bw_cell); in Prepare()
336 NN_CHECK_EQ(aux_input_->shape().dimensions[0], input_->shape().dimensions[0]); in Prepare()
337 NN_CHECK_EQ(aux_input_->shape().dimensions[1], input_->shape().dimensions[1]); in Prepare()
DQuantizedLSTM.cpp315 NN_CHECK_EQ(NumDimensions(prevCellState), 2); in prepare()
316 NN_CHECK_EQ(SizeOfDimension(prevCellState, 0), numBatches); in prepare()
317 NN_CHECK_EQ(SizeOfDimension(prevCellState, 1), outputSize); in prepare()
318 NN_CHECK_EQ(prevCellState->zeroPoint, 0); in prepare()
DMultinomial.cpp68 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 3); in Prepare()
69 NN_CHECK_EQ(NumOutputs(operation), 1); in Prepare()
/packages/modules/NeuralNetworks/common/include/
DOperationsUtils.h35 #define NN_CHECK_EQ(x, y) NN_RET_CHECK_EQ(x, y) macro