Home
last modified time | relevance | path

Searched defs:IS_TRAINING (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op_test.cc309 #define BM_FusedBatchNorm(N, H, W, C, T, IS_TRAINING, FORMAT, DEVICE) \ argument
349 #define BM_FusedBatchNormGrad(N, H, W, C, T, IS_TRAINING, FORMAT, DEVICE) \ argument
363 #define BM_FusedBatchNormGradResnetShapes(T, IS_TRAINING, FORMAT, DEVICE) \ argument
Dfused_batch_norm_ex_op_test.cc622 #define BM_NAME(N, H, W, C, T, FORMAT, IS_TRAINING, A, DEVICE) \ argument
626 #define BM_FusedBatchNorm(N, H, W, C, T, FORMAT, IS_TRAINING, ACTIVATION, \ argument