Lines Matching refs:NumDimensions
159 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()
235 NN_CHECK_EQ(NumDimensions(forget_gate_bias), 1); in CheckInputTensorDimensions()
238 NN_CHECK_EQ(NumDimensions(cell_bias), 1); 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()
265 NN_CHECK_EQ(NumDimensions(input_layer_norm_weights), 1); in CheckInputTensorDimensions()
269 NN_CHECK_EQ(NumDimensions(forget_layer_norm_weights), 1); in CheckInputTensorDimensions()
273 NN_CHECK_EQ(NumDimensions(cell_layer_norm_weights), 1); in CheckInputTensorDimensions()
277 NN_CHECK_EQ(NumDimensions(output_layer_norm_weights), 1); in CheckInputTensorDimensions()
347 NN_CHECK(NumDimensions(input_) > 1); in Prepare()
352 NN_CHECK_EQ(NumDimensions(input_to_output_weights_), 2); in Prepare()
355 NN_CHECK_EQ(NumDimensions(recurrent_to_output_weights_), 2); in Prepare()