| /*! |
| * Copyright (c) 2017 by Contributors |
| * \brief Tag definitions |
| * \file tags.h |
| */ |
| #ifndef TOPI_TAGS_H_ |
| #define TOPI_TAGS_H_ |
| |
| #include <string> |
| |
| namespace topi { |
| |
| constexpr auto kElementWise = "elemwise"; |
| constexpr auto kInjective = "injective"; |
| constexpr auto kCommReduce = "comm_reduce"; |
| constexpr auto kCommReduceIdx = "comm_reduce_idx"; |
| constexpr auto kBroadcast = "broadcast"; |
| constexpr auto kMatMul = "matmul"; |
| constexpr auto kConv2dNCHW = "conv2d_nchw"; |
| constexpr auto kConv2dHWCN = "conv2d_hwcn"; |
| constexpr auto kDepthwiseConv2dNCHW = "depthwise_conv2d_nchw"; |
| constexpr auto kDepthwiseConv2dNHWC = "depthwise_conv2d_nhwc"; |
| constexpr auto kDepthwiseConv2dBackInputNHWC = "depthwise_conv2d_back_input_nhwc"; |
| constexpr auto kDepthwiseConv2dBackWeightNHWC = "depthwise_conv2d_back_weight_nhwc"; |
| constexpr auto kGroupConv2d = "group_conv2d"; |
| |
| inline bool is_broadcast(std::string tag) { |
| return |
| tag.rfind(kElementWise, 0) == 0 || |
| tag.rfind(kBroadcast, 0) == 0; |
| } |
| |
| inline bool is_injective(std::string tag) { |
| return |
| tag.rfind(kElementWise, 0) == 0 || |
| tag.rfind(kBroadcast, 0) == 0 || |
| tag.rfind(kInjective, 0) == 0; |
| } |
| |
| } // namespace topi |
| |
| #endif // TOPI_TAGS_H_ |
