Home
last modified time | relevance | path

Searched refs:GET_FLAG (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc27 #define GET_FLAG(bn_flag) static_cast<int>(mkldnn::normalization_flags::bn_flag) macro
28 #define IS_SET(cflag) (context_.flags & GET_FLAG(cflag))
188 ? GET_FLAG(use_scale_shift) in Setup()
189 : (GET_FLAG(use_scale_shift) | GET_FLAG(use_global_stats)); in Setup()
194 context_.flags |= GET_FLAG(fuse_norm_relu); in Setup()
248 if ((IS_SET(use_scale_shift)) && GET_FLAG(use_scale_shift)) { in Setup()
283 if ((IS_SET(use_scale_shift)) && GET_FLAG(use_scale_shift)) { in Setup()
522 ? GET_FLAG(use_scale_shift) in Setup()
523 : (GET_FLAG(use_scale_shift) | GET_FLAG(use_global_stats)); in Setup()
1534 #undef GET_FLAG