Home
last modified time | relevance | path

Searched defs:feature_idx (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DEmbeddingBag.cu493 for (int feature_idx = thread_in_warp; feature_idx < embedding_features; in _embedding_bag_per_sample_weights_backward_kernel() local
/external/libaom/av1/encoder/
Dpartition_search.c5925 int feature_idx = 0; in ml_predict_var_partitioning() local
6003 int feature_idx = 0; in store_partition_data() local
Dtx_search.c1660 int feature_idx = 2; in get_mean_dev_features() local
2804 int feature_idx = get_mean_dev_features(diff, diff_stride, bw, bh, features); in ml_predict_intra_tx_depth_prune() local
Dpartition_strategy.c1008 int *feature_idx) { in add_rd_feature()
/external/libvpx/vp9/encoder/
Dvp9_encodeframe.c3561 int feature_idx = 0; in ml_predict_var_rd_partitioning() local
4877 int feature_idx = 0; in ml_predict_var_partitioning() local