tags.h File Reference

External function interface to rocBLAS libraries. More...

#include <string>

Include dependency graph for tags.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::topi

Functions
bool	tvm::topi::is_broadcast (std::string tag)
bool	tvm::topi::is_injective (std::string tag)

Variables
constexpr auto	tvm::topi::kElementWise = "elemwise"
constexpr auto	tvm::topi::kInjective = "injective"
constexpr auto	tvm::topi::kCommReduce = "comm_reduce"
constexpr auto	tvm::topi::kCommReduceIdx = "comm_reduce_idx"
constexpr auto	tvm::topi::kBroadcast = "broadcast"
constexpr auto	tvm::topi::kMatMul = "matmul"
constexpr auto	tvm::topi::kConv2dNCHW = "conv2d_nchw"
constexpr auto	tvm::topi::kConv2dHWCN = "conv2d_hwcn"
constexpr auto	tvm::topi::kDepthwiseConv2dNCHW = "depthwise_conv2d_nchw"
constexpr auto	tvm::topi::kDepthwiseConv2dNHWC = "depthwise_conv2d_nhwc"
constexpr auto	tvm::topi::kDepthwiseConv2dBackInputNHWC = "depthwise_conv2d_back_input_nhwc"
constexpr auto	tvm::topi::kDepthwiseConv2dBackWeightNHWC = "depthwise_conv2d_back_weight_nhwc"
constexpr auto	tvm::topi::kEinsum = "einsum"
constexpr auto	tvm::topi::kGroupConv2d = "group_conv2d"

Detailed Description

External function interface to rocBLAS libraries.

Tag definitions.