bnn.h File Reference

Binary op constructions. More...

#include <tvm/arith/analyzer.h>
#include <tvm/te/operation.h>
#include <tvm/topi/detail/constant_utils.h>
#include <tvm/topi/tags.h>
#include <string>

Include dependency graph for bnn.h:

Go to the source code of this file.

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

Functions
tvm::te::Tensor	tvm::topi::nn::binarize_pack (const tvm::te::Tensor &data, int axis, std::string name="PackedInput", std::string tag="binarize_pack")
	Binarization and bit-packing along a certain axis. More...
tvm::te::Tensor	tvm::topi::nn::binary_dense (const tvm::te::Tensor &data, const tvm::te::Tensor &weight)
	Binary matrix multiplication using xor and bit-count. More...

Detailed Description

Binary op constructions.