Searched refs:input_box_encodings (Results 1 – 1 of 1) sorted by relevance
144 const TfLiteTensor* input_box_encodings = in Prepare() local150 TF_LITE_ENSURE_EQ(context, NumDimensions(input_box_encodings), 3); in Prepare()198 {input_box_encodings->dims->data[1], kNumCoordBox}); in Prepare()214 {input_box_encodings->dims->data[1]}); in Prepare()232 void DequantizeBoxEncodings(const TfLiteTensor* input_box_encodings, int idx, in DequantizeBoxEncodings() argument236 GetTensorData<uint8>(input_box_encodings) + kNumCoordBox * idx; in DequantizeBoxEncodings()261 const TfLiteTensor* input_box_encodings = in DecodeCenterSizeBoxes() local263 TF_LITE_ENSURE_EQ(context, input_box_encodings->dims->data[0], kBatchSize); in DecodeCenterSizeBoxes()264 const int num_boxes = input_box_encodings->dims->data[1]; in DecodeCenterSizeBoxes()265 TF_LITE_ENSURE_EQ(context, input_box_encodings->dims->data[2], kNumCoordBox); in DecodeCenterSizeBoxes()[all …]