Searched defs:softmax_dim_size (Results 1 – 1 of 1) sorted by relevance
1085 auto softmax_dim_size = a.sizes()[softmax_dim]; in TEST_F() local1158 auto softmax_dim_size = a.sizes()[softmax_dim]; in TEST_F() local1239 auto softmax_dim_size = a.sizes()[softmax_dim]; in TEST_F() local