Home
last modified time | relevance | path

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

/frameworks/ml/nn/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.cpp53 NN_CHECK_EQ(NumOutputs(operation), 1); in Prepare()
56 NN_CHECK_EQ(NumDimensions(hash), 2); in Prepare()
74 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 4); in Prepare()
75 NN_CHECK_EQ(NumDimensions(weight), 1); in Prepare()
76 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, 0); 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.cpp55 NN_CHECK_EQ(NumOutputs(operation), 2); in Prepare()
67 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(input_weights, 1)); in Prepare()
68 NN_CHECK_EQ(SizeOfDimension(input_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
69 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
70 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 1), SizeOfDimension(bias, 0)); in Prepare()
DMultinomial.cpp66 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 3); in Prepare()
67 NN_CHECK_EQ(NumOutputs(operation), 1); in Prepare()
DBidirectionalSequenceLSTM.cpp274 NN_CHECK_EQ(NumDimensions(fw_input_to_output_weights_), 2); in Prepare()
275 NN_CHECK_EQ(SizeOfDimension(fw_input_to_output_weights_, 1), n_fw_input); in Prepare()
277 NN_CHECK_EQ(NumDimensions(fw_recurrent_to_output_weights_), 2); in Prepare()
278 NN_CHECK_EQ(SizeOfDimension(fw_recurrent_to_output_weights_, 0), n_fw_cell); in Prepare()
283 NN_CHECK_EQ(NumDimensions(bw_recurrent_to_output_weights_), 2); in Prepare()
284 NN_CHECK_EQ(SizeOfDimension(bw_recurrent_to_output_weights_, 0), n_bw_cell); in Prepare()
337 NN_CHECK_EQ(aux_input_->shape().dimensions[0], input_->shape().dimensions[0]); in Prepare()
338 NN_CHECK_EQ(aux_input_->shape().dimensions[1], input_->shape().dimensions[1]); in Prepare()
DQuantizedLSTM.cpp318 NN_CHECK_EQ(NumDimensions(prevCellState), 2); in prepare()
319 NN_CHECK_EQ(SizeOfDimension(prevCellState, 0), numBatches); in prepare()
320 NN_CHECK_EQ(SizeOfDimension(prevCellState, 1), outputSize); in prepare()
321 NN_CHECK_EQ(prevCellState->zeroPoint, 0); in prepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h35 #define NN_CHECK_EQ(x, y) NN_RET_CHECK_EQ(x, y) macro