Home
last modified time | relevance | path

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

/hardware/google/pixel/thermal/virtualtemp_estimator/
Dvirtualtemp_estimator.cpp270 size_t output_label_count = data.output_label_count; in TFliteInitialize() local
298 if (output_label_count < 1 || num_hot_spots < 1) { in TFliteInitialize()
301 << " predicted sample order: " << output_label_count; in TFliteInitialize()
305 tflite_instance_->output_label_count = output_label_count; in TFliteInitialize()
307 tflite_instance_->output_buffer_size = output_label_count * num_hot_spots; in TFliteInitialize()
613 size_t output_label_count = tflite_instance_->output_label_count; in TFlitePredictAfterTimeMs() local
616 if (request_step == output_label_count - 1) { in TFlitePredictAfterTimeMs()
618 prediction = output_buffer[output_label_count - 1]; in TFlitePredictAfterTimeMs()
750 sample_interval_ms * (tflite_instance_->output_label_count - 1); in ParseInputConfig()
Dvirtualtemp_estimator.h44 size_t output_label_count; member
66 ml_model_init_data.output_label_count = 1; in VtEstimationInitData()
Dvirtualtemp_estimator_data.h83 output_label_count = 1; in VtEstimatorTFLiteData()
109 size_t output_label_count; member
/hardware/google/pixel/thermal/utils/
Dthermal_info.cpp452 init_data.ml_model_init_data.output_label_count = sensor["OutputLabelCount"].asInt(); in ParseVirtualSensorInfo()
454 << init_data.ml_model_init_data.output_label_count << " labels"; in ParseVirtualSensorInfo()