File size: 4,451 Bytes
d3a8b6d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
-------------------------------------------  ------------  ------------  ------------  ------------  ------------  ------------  
                                       Name    Self CPU %      Self CPU   CPU total %     CPU total  CPU time avg    # of Calls  
-------------------------------------------  ------------  ------------  ------------  ------------  ------------  ------------  
                    aten::cudnn_convolution        69.86%     444.597ms        69.86%     444.597ms      37.050ms            12  
                         aten::_log_softmax         7.52%      47.831ms         7.52%      47.831ms      47.831ms             1  
                            aten::clamp_min         4.42%      28.104ms         4.42%      28.104ms       3.513ms             8  
                     aten::cudnn_batch_norm         4.13%      26.264ms         4.20%      26.758ms       2.676ms            10  
                                 aten::add_         3.47%      22.086ms         3.47%      22.086ms       2.209ms            10  
                           aten::bernoulli_         2.79%      17.777ms         2.79%      17.777ms       2.222ms             8  
                                 aten::div_         2.53%      16.126ms         2.53%      16.126ms       2.016ms             8  
                                  aten::mul         2.29%      14.584ms         2.29%      14.584ms       1.823ms             8  
                           aten::avg_pool2d         0.63%       4.009ms         0.63%       4.009ms       4.009ms             1  
              aten::max_pool2d_with_indices         0.54%       3.446ms         0.54%       3.446ms       1.149ms             3  
                          aten::convolution         0.39%       2.469ms        70.31%     447.487ms      37.291ms            12  
                                 aten::relu         0.28%       1.804ms         4.70%      29.908ms       3.739ms             8  
               aten::_batch_norm_impl_index         0.22%       1.430ms         4.43%      28.188ms       2.819ms            10  
                           aten::batch_norm         0.16%       1.006ms         4.59%      29.194ms       2.919ms            10  
                                aten::empty         0.12%     757.000us         0.12%     757.000us      11.828us            64  
                           aten::max_pool2d         0.12%     751.000us         0.66%       4.197ms       1.399ms             3  
                          aten::log_softmax         0.10%     653.000us         7.62%      48.484ms      48.484ms             1  
                               aten::conv2d         0.10%     636.000us        70.41%     448.123ms      37.344ms            12  
                      aten::feature_dropout         0.08%     479.000us         7.71%      49.058ms       6.132ms             8  
                                aten::copy_         0.07%     447.000us         0.07%     447.000us      63.857us             7  
                         aten::_convolution         0.07%     421.000us        69.92%     445.018ms      37.085ms            12  
                                   aten::to         0.05%     291.000us         0.13%     843.000us       7.205us           117  
                                aten::zeros         0.04%     270.000us         0.08%     523.000us      87.167us             6  
                        aten::empty_strided         0.01%      60.000us         0.01%      60.000us       8.571us             7  
                             aten::_to_copy         0.01%      45.000us         0.09%     552.000us      78.857us             7  
                                 aten::view         0.01%      39.000us         0.01%      39.000us       3.545us            11  
                           aten::empty_like         0.00%      31.000us         0.06%     384.000us      38.400us            10  
                            aten::new_empty         0.00%      23.000us         0.01%      92.000us      11.500us             8  
    aten::_has_compatible_shallow_copy_type         0.00%       2.000us         0.00%       2.000us       0.031us            64  
                                aten::zero_         0.00%       1.000us         0.00%       1.000us       0.167us             6  
-------------------------------------------  ------------  ------------  ------------  ------------  ------------  ------------  
Self CPU time total: 636.439ms