Searched refs:pred_probs (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 45 self.pred_probs = []
50 self.pred_probs = []
56 self.pred_probs.extend(batch_predict.flatten().tolist())
60 if len(self.true_labels) != len(self.pred_probs):
62 auc = roc_auc_score(self.true_labels, self.pred_probs)
338 pred_probs = self.sigmoid(logits)
340 return logits, pred_probs, labels
|
| /third_party/gstreamer/gstplugins_bad/sys/d3d11/ |
| D | gstd3d11vp9dec.cpp | 645 G_STATIC_ASSERT (G_N_ELEMENTS (params->stVP9Segments.pred_probs) == in gst_d3d11_vp9_dec_copy_segmentation_params()
647 G_STATIC_ASSERT (sizeof (params->stVP9Segments.pred_probs) == in gst_d3d11_vp9_dec_copy_segmentation_params()
651 memcpy (params->stVP9Segments.pred_probs, sp->segmentation_pred_prob, in gst_d3d11_vp9_dec_copy_segmentation_params()
652 sizeof (params->stVP9Segments.pred_probs)); in gst_d3d11_vp9_dec_copy_segmentation_params()
654 memset (params->stVP9Segments.pred_probs, 255, in gst_d3d11_vp9_dec_copy_segmentation_params()
655 sizeof (params->stVP9Segments.pred_probs)); in gst_d3d11_vp9_dec_copy_segmentation_params()
|
| /third_party/ffmpeg/libavcodec/ |
| D | dxva2_vp9.c | 129 pp->stVP9Segments.pred_probs[i] = h->h.segmentation.pred_prob[i]; in fill_picture_parameters()
131 memset(pp->stVP9Segments.pred_probs, 255, sizeof(pp->stVP9Segments.pred_probs)); in fill_picture_parameters()
|
| /third_party/gstreamer/gstplugins_bad/gst-libs/gst/codecparsers/ |
| D | gstvp9parser.c | 298 seg->pred_probs[i] = seg->update_pred_probs[i] ? in parse_segmentation()
303 seg->pred_probs[i] = GST_VP9_MAX_PROB; in parse_segmentation()
457 G_N_ELEMENTS (info->pred_probs)); in segmentation_save()
460 memcpy (parser->segment_pred_probs, info->pred_probs, in segmentation_save()
461 sizeof (info->pred_probs)); in segmentation_save()
|
| D | gstvp9parser.h | 337 guint8 pred_probs[GST_VP9_PREDICTION_PROBS]; member
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | wide_and_deep.py | 327 pred_probs = self.sigmoid(logits)
328 return logits, pred_probs, labels
|
| /third_party/gstreamer/gstplugins_bad/sys/v4l2codecs/ |
| D | gstv4l2codecvp9dec.c | 143 G_STATIC_ASSERT (sizeof (v4l2_segmentation->pred_probs) == in gst_v4l2_codec_vp9_dec_fill_seg_params()
151 for (i = 0; i < G_N_ELEMENTS (v4l2_segmentation->pred_probs); i++) { in gst_v4l2_codec_vp9_dec_fill_seg_params()
152 v4l2_segmentation->pred_probs[i] = s->segmentation_pred_prob[i]; in gst_v4l2_codec_vp9_dec_fill_seg_params()
|
| /third_party/mesa3d/src/gallium/drivers/radeon/ |
| D | radeon_vcn_dec.h | 1036 unsigned char pred_probs[3]; member
|
| D | radeon_vcn_dec.c | 413 prbs->seg.pred_probs[i] = pic->picture_parameter.segment_pred_probs[i]; in get_vp9_msg()
|
| /third_party/gstreamer/gstplugins_bad/sys/v4l2codecs/linux/ |
| D | v4l2-controls.h | 2109 __u8 pred_probs[3]; member
|
| /third_party/gstreamer/gstplugins_good/sys/v4l2/ext/ |
| D | v4l2-controls.h | 2109 __u8 pred_probs[3]; member
|