D | nn_ops_test.cc | 876 #define BM_AvgPoolFwdCPU(BS, IR, IC, ND, KR, KC, ST, PT, TH, LABEL) \ argument 982 #define BM_AvgPoolBkCPU(BS, IR, IC, ND, KR, KC, ST, PT, TH, LABEL) \ argument 1071 #define BM_MaxPoolFwdCPU(BS, IR, IC, ND, KR, KC, ST, PT, TH, LABEL) \ argument 1149 #define BM_MaxPoolBkGPU(BS, IR, IC, ND, KR, KC, ST, PT, TH, LABEL) \ argument 1160 #define BM_MaxPoolBkCPU(BS, IR, IC, ND, KR, KC, ST, PT, TH, LABEL) \ argument 1242 #define BM_Relu(BS, IR, IC, ND, TH, LABEL) \ argument 1337 #define BM_TopKGPU(IR, IC, IK, TH, LABEL) \ argument 1343 #define BM_TopKCPU(IR, IC, IK, TH, LABEL) \ argument
|