Home
last modified time | relevance | path

Searched refs:TF_LITE_ASSERT_EQ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc470 TF_LITE_ASSERT_EQ(sizeof(uint8_t), 1); in TfLiteTypeGetSize()
473 TF_LITE_ASSERT_EQ(sizeof(int8_t), 1); in TfLiteTypeGetSize()
478 TF_LITE_ASSERT_EQ(sizeof(int16_t), 2); in TfLiteTypeGetSize()
481 TF_LITE_ASSERT_EQ(sizeof(int16_t), 2); in TfLiteTypeGetSize()
484 TF_LITE_ASSERT_EQ(sizeof(float), 4); in TfLiteTypeGetSize()
487 TF_LITE_ASSERT_EQ(sizeof(int32_t), 4); in TfLiteTypeGetSize()
490 TF_LITE_ASSERT_EQ(sizeof(uint32_t), 4); in TfLiteTypeGetSize()
493 TF_LITE_ASSERT_EQ(sizeof(int64_t), 8); in TfLiteTypeGetSize()
496 TF_LITE_ASSERT_EQ(sizeof(uint64_t), 8); in TfLiteTypeGetSize()
499 TF_LITE_ASSERT_EQ(sizeof(double), 8); in TfLiteTypeGetSize()
[all …]
Dop_macros.h78 #define TF_LITE_ASSERT_EQ(x, y) \ macro
Dbidirectional_sequence_rnn.cc176 TF_LITE_ASSERT_EQ(aux_input->dims->data[0], input->dims->data[0]); in Prepare()
177 TF_LITE_ASSERT_EQ(aux_input->dims->data[1], input->dims->data[1]); in Prepare()
180 TF_LITE_ASSERT_EQ(fw_aux_input_weights->dims->data[0], fw_num_units); in Prepare()
181 TF_LITE_ASSERT_EQ(bw_aux_input_weights->dims->data[0], bw_num_units); in Prepare()
182 TF_LITE_ASSERT_EQ(aux_input->dims->data[2], in Prepare()
184 TF_LITE_ASSERT_EQ(aux_input->dims->data[2], in Prepare()
Dbidirectional_sequence_lstm.cc527 TF_LITE_ASSERT_EQ(aux_input->dims->data[0], input->dims->data[0]); in Prepare()
528 TF_LITE_ASSERT_EQ(aux_input->dims->data[1], input->dims->data[1]); in Prepare()