Home
last modified time | relevance | path

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 Dbias_add.h46 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 Dbias_add.h46 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 Dbias_add.h46 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 Dcanonicalize_ops.cc37 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 Dcanonicalize_ops.cc37 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 Dauto_scheduler_test.cc49 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 Dethosn_api.cc45 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__.py150 def bias_add(value, bias): function
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/
H A Dnn.py552 def bias_add(data, bias, axis=1): function
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/
H A Dnn.py552 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 Dnn.py1356 def bias_add(data, bias, axis=1): function