Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTM.cpp164 NN_CHECK_EQ(NumDimensions(input_to_forget_weights), 2); in CheckInputTensorDimensions()
168 NN_CHECK_EQ(NumDimensions(input_to_cell_weights), 2); in CheckInputTensorDimensions()
182 NN_CHECK_EQ(NumDimensions(recurrent_to_cell_weights), 2); in CheckInputTensorDimensions()
231 NN_CHECK_EQ(NumDimensions(input_gate_bias), 1); in CheckInputTensorDimensions()
235 NN_CHECK_EQ(NumDimensions(forget_gate_bias), 1); in CheckInputTensorDimensions()
238 NN_CHECK_EQ(NumDimensions(cell_bias), 1); in CheckInputTensorDimensions()
239 NN_CHECK_EQ(SizeOfDimension(cell_bias, 0), n_cell); in CheckInputTensorDimensions()
241 NN_CHECK_EQ(NumDimensions(output_gate_bias), 1); in CheckInputTensorDimensions()
245 NN_CHECK_EQ(NumDimensions(projection_weights), 2); in CheckInputTensorDimensions()
251 NN_CHECK_EQ(NumDimensions(projection_bias), 1); in CheckInputTensorDimensions()
[all …]
H A 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()
H A 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()
H A 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()
H A 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()
H A DMultinomial.cpp68 NN_CHECK_EQ(NumInputsWithValues(operation, operands), 3); in Prepare()
69 NN_CHECK_EQ(NumOutputs(operation), 1); in Prepare()
H A 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()
/aosp12/packages/modules/NeuralNetworks/common/include/
H A DOperationsUtils.h35 #define NN_CHECK_EQ(x, y) NN_RET_CHECK_EQ(x, y) macro