Searched defs:feature_idx (Results 1 – 5 of 5) sorted by relevance
493 for (int feature_idx = thread_in_warp; feature_idx < embedding_features; in _embedding_bag_per_sample_weights_backward_kernel() local
5925 int feature_idx = 0; in ml_predict_var_partitioning() local6003 int feature_idx = 0; in store_partition_data() local
1660 int feature_idx = 2; in get_mean_dev_features() local2804 int feature_idx = get_mean_dev_features(diff, diff_stride, bw, bh, features); in ml_predict_intra_tx_depth_prune() local
1008 int *feature_idx) { in add_rd_feature()
3561 int feature_idx = 0; in ml_predict_var_rd_partitioning() local4877 int feature_idx = 0; in ml_predict_var_partitioning() local