Searched defs:bias_add (Results 1 – 11 of 11) sorted by relevance
/dports/misc/tvm/incubator-tvm-0.6.1/topi/include/topi/nn/ |
H A D | bias_add.h | 46 inline tvm::Tensor bias_add(const tvm::Tensor& data, const tvm::Tensor& bias, int axis) { in bias_add() function
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/topi/nn/ |
H A D | bias_add.h | 46 inline tvm::te::Tensor bias_add(const tvm::te::Tensor& data, const tvm::te::Tensor& bias, in bias_add() function
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/include/topi/nn/ |
H A D | bias_add.h | 46 inline tvm::Tensor bias_add(const tvm::Tensor& data, const tvm::Tensor& bias, int axis) { in bias_add() function
|
/dports/misc/tvm/incubator-tvm-0.6.1/src/relay/pass/ |
H A D | canonicalize_ops.cc | 37 static const Op& bias_add = Op::Get("nn.bias_add"); in VisitExpr_() local
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/pass/ |
H A D | canonicalize_ops.cc | 37 static const Op& bias_add = Op::Get("nn.bias_add"); in VisitExpr_() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/cpp/ |
H A D | auto_scheduler_test.cc | 49 const auto& bias_add = compute( in conv2d_nchw_bn_relu_func() local 73 int data = 0, padding = 1, kernel = 2, conv = 3, bias = 4, bias_add = 5; in TEST() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/backend/contrib/ethosn/ |
H A D | ethosn_api.cc | 45 Call bias_add = Downcast<Call>(requantize->args[0]); in QnnConv2d() local 144 Call bias_add = Downcast<Call>(requantize->args[0]); in QnnFullyConnected() local
|
/dports/misc/mnn/MNN-1.2.0/pymnn/pip_package/MNN/expr/ |
H A D | __init__.py | 150 def bias_add(value, bias): function
|
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/ |
H A D | nn.py | 552 def bias_add(data, bias, axis=1): function
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/ |
H A D | nn.py | 552 def bias_add(data, bias, axis=1): function
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/nn/ |
H A D | nn.py | 1356 def bias_add(data, bias, axis=1): function
|