Searched defs:ACTIVATION (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cu.cc | 340 #define LAUNCH(DATA_FORMAT, ADD_SIDE_INPUT, ACTIVATION, CHANNEL_SIZE, \ in operator ()() argument
|
| D | fused_batch_norm_ex_op_test.cc | 605 #define BM_FusedBatchNorm(N, H, W, C, T, FORMAT, IS_TRAINING, ACTIVATION, \ argument
|