Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTM.cpp159 NN_CHECK_EQ(NumDimensions(input_to_input_weights), 2u); 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), 2u); 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), 2u); 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), 2u); in CheckInputTensorDimensions()
[all …]
DLSHProjection.cpp52 NN_CHECK_EQ(NumOutputs(operation), 1); in Prepare()
55 NN_CHECK_EQ(NumDimensions(hash), 2u); in Prepare()
73 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 4); in Prepare()
74 NN_CHECK_EQ(NumDimensions(weight), 1u); in Prepare()
75 NN_CHECK_EQ(SizeOfDimension(weight, 0), SizeOfDimension(input, 0)); in Prepare()
DSVDF.cpp66 NN_CHECK_EQ(NumOutputs(operation), 2); in Prepare()
85 NN_CHECK_EQ(num_filters % rank, 0u); in Prepare()
88 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(weights_feature, 1)); in Prepare()
89 NN_CHECK_EQ(SizeOfDimension(weights_time, 0), num_filters); in Prepare()
93 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_), 2u); 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_), 2u); 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_), 2u); 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()
DMultinomial.cpp72 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 3); in Prepare()
73 NN_CHECK_EQ(NumOutputs(operation), 1); in Prepare()
DQuantizedLSTM.cpp319 NN_CHECK_EQ(NumDimensions(prevCellState), 2u); in prepare()
320 NN_CHECK_EQ(SizeOfDimension(prevCellState, 0), numBatches); in prepare()
321 NN_CHECK_EQ(SizeOfDimension(prevCellState, 1), outputSize); in prepare()
322 NN_CHECK_EQ(prevCellState->zeroPoint, 0); in prepare()
/packages/modules/NeuralNetworks/common/types/include/
DOperationsUtils.h33 #define NN_CHECK_EQ(x, y) NN_RET_CHECK_EQ(x, y) macro