Home
last modified time | relevance | path

Searched refs:nnvm (Results 1 – 25 of 1392) sorted by relevance

12345678910>>...56

/dports/misc/tvm/incubator-tvm-0.6.1/docs/
H A Dnnvm_top.rst48 nnvm.symbol.relu
50 nnvm.symbol.tanh
52 nnvm.symbol.exp
53 nnvm.symbol.log
54 nnvm.symbol.sqrt
74 nnvm.symbol.pad
82 nnvm.symbol.lrn
114 nnvm.symbol.abs
135 nnvm.symbol.sum
136 nnvm.symbol.min
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/docs/
H A Dnnvm_top.rst48 nnvm.symbol.relu
50 nnvm.symbol.tanh
52 nnvm.symbol.exp
53 nnvm.symbol.log
54 nnvm.symbol.sqrt
74 nnvm.symbol.pad
82 nnvm.symbol.lrn
114 nnvm.symbol.abs
135 nnvm.symbol.sum
136 nnvm.symbol.min
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Delemwise_unary_op_trig.cc52 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40102()
62 {nnvm::NodeEntry{ in __anoncc59c6b40102()
95 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40202()
145 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40302()
193 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40402()
239 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40502()
288 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40602()
376 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40702()
414 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40802()
456 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anoncc59c6b40902()
[all …]
H A Dla_op.cc113 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
184 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
235 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
294 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
356 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
419 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
462 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
513 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
565 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
623 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
[all …]
H A Delemwise_unary_op_pow.cc50 .set_attr<nnvm::FGradient>("FGradient",
51 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbe9d95d40102()
73 std::vector<nnvm::NodeEntry> ret; in __anonbe9d95d40102()
125 .set_attr<nnvm::FGradient>("FGradient",
126 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbe9d95d40202()
146 std::vector<nnvm::NodeEntry> ret; in __anonbe9d95d40202()
175 .set_attr<nnvm::FGradient>("FGradient",
176 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbe9d95d40302()
227 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbe9d95d40402()
276 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbe9d95d40502()
[all …]
H A Delemwise_unary_op_basic.cc90 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonb5e792e00102()
126 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonb5e792e00202()
136 {nnvm::NodeEntry{ones}, nnvm::NodeEntry{two_y}}, nullptr, &n); in __anonb5e792e00202()
371 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonb5e792e00f02()
385 .set_attr<nnvm::FInplaceOption>(
398 .set_attr<nnvm::FGradient>(
517 .set_attr<nnvm::FInplaceOption>(
527 .set_attr<nnvm::FInferType>("FInferType", [](const nnvm::NodeAttrs& attrs,
537 .set_attr<nnvm::FGradient>(
579 [](const nnvm::NodeAttrs& attrs,
[all …]
H A Delemwise_unary_op_logexp.cc111 .set_attr<nnvm::FGradient>("FGradient",
112 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon47dfd42d0102()
127 std::vector<nnvm::NodeEntry> ret; in __anon47dfd42d0102()
137 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon47dfd42d0202()
153 std::vector<nnvm::NodeEntry> ret; in __anon47dfd42d0202()
163 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon47dfd42d0302()
179 std::vector<nnvm::NodeEntry> ret; in __anon47dfd42d0302()
205 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon47dfd42d0402()
225 std::vector<nnvm::NodeEntry> ret; in __anon47dfd42d0402()
249 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon47dfd42d0502()
[all …]
H A Dmatrix_op.cc43 nnvm::dim_t begin_row = begin[0], end_row = end[0]; in SliceDimTwoCsrImpl()
44 nnvm::dim_t begin_col = begin[1], end_col = end[1]; in SliceDimTwoCsrImpl()
45 nnvm::dim_t indptr_len = end_row - begin_row + 1; in SliceDimTwoCsrImpl()
190 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
264 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
333 .set_attr<nnvm::FGradient>("FGradient",
334 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonb9889e8b0702()
400 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
421 void SliceExCPU(const nnvm::NodeAttrs& attrs, in SliceExCPU()
504 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Delemwise_unary_op_trig.cc52 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0102()
62 {nnvm::NodeEntry{ in __anon9c5cffde0102()
95 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0202()
145 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0302()
193 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0402()
239 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0502()
288 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0602()
376 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0702()
414 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0802()
456 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon9c5cffde0902()
[all …]
H A Dla_op.cc113 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
184 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
235 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
294 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
356 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
419 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
462 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
513 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
565 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
623 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
[all …]
H A Delemwise_unary_op_pow.cc50 .set_attr<nnvm::FGradient>("FGradient",
51 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbd2951be0102()
73 std::vector<nnvm::NodeEntry> ret; in __anonbd2951be0102()
125 .set_attr<nnvm::FGradient>("FGradient",
126 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbd2951be0202()
146 std::vector<nnvm::NodeEntry> ret; in __anonbd2951be0202()
175 .set_attr<nnvm::FGradient>("FGradient",
176 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbd2951be0302()
227 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbd2951be0402()
276 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonbd2951be0502()
[all …]
H A Delemwise_unary_op_basic.cc90 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon8651f14a0102()
126 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon8651f14a0202()
136 {nnvm::NodeEntry{ones}, nnvm::NodeEntry{two_y}}, nullptr, &n); in __anon8651f14a0202()
371 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon8651f14a0f02()
385 .set_attr<nnvm::FInplaceOption>(
398 .set_attr<nnvm::FGradient>(
517 .set_attr<nnvm::FInplaceOption>(
527 .set_attr<nnvm::FInferType>("FInferType", [](const nnvm::NodeAttrs& attrs,
537 .set_attr<nnvm::FGradient>(
579 [](const nnvm::NodeAttrs& attrs,
[all …]
H A Delemwise_unary_op_logexp.cc111 .set_attr<nnvm::FGradient>("FGradient",
112 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon2595ffd70102()
127 std::vector<nnvm::NodeEntry> ret; in __anon2595ffd70102()
137 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon2595ffd70202()
153 std::vector<nnvm::NodeEntry> ret; in __anon2595ffd70202()
163 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon2595ffd70302()
179 std::vector<nnvm::NodeEntry> ret; in __anon2595ffd70302()
205 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon2595ffd70402()
225 std::vector<nnvm::NodeEntry> ret; in __anon2595ffd70402()
249 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon2595ffd70502()
[all …]
H A Dmatrix_op.cc43 nnvm::dim_t begin_row = begin[0], end_row = end[0]; in SliceDimTwoCsrImpl()
44 nnvm::dim_t begin_col = begin[1], end_col = end[1]; in SliceDimTwoCsrImpl()
45 nnvm::dim_t indptr_len = end_row - begin_row + 1; in SliceDimTwoCsrImpl()
190 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
264 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
333 .set_attr<nnvm::FGradient>("FGradient",
334 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonc643ab750702()
400 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
421 void SliceExCPU(const nnvm::NodeAttrs& attrs, in SliceExCPU()
504 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_elemwise_unary_op_basic.cc52 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
57 .set_attr<nnvm::FInplaceIdentity>("FInplaceIdentity",
62 .set_attr<nnvm::FListInputNames>("FListInputNames",
183 .set_attr<nnvm::FListInputNames>("FListInputNames",
252 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
256 .set_attr<nnvm::FListInputNames>("FListInputNames",
446 .set_attr<nnvm::FListInputNames>("FListInputNames",
453 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
468 .set_attr<nnvm::FListInputNames>("FListInputNames",
476 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
[all …]
H A Dnp_elemwise_broadcast_op_extended.cc54 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
55 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
69 .set_attr<nnvm::FListInputNames>("FListInputNames",
75 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
90 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
108 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
123 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
141 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
159 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
174 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
[all …]
H A Dnp_where_op.cc104 .set_attr<nnvm::FGradient>("FGradient",
108 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon1ce6144c0302()
109 std::vector<nnvm::NodeEntry> ret; in __anon1ce6144c0302()
118 p = nnvm::Node::Create(); in __anon1ce6144c0302()
164 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon1ce6144c0702()
165 std::vector<nnvm::NodeEntry> ret; in __anon1ce6144c0702()
174 p = nnvm::Node::Create(); in __anon1ce6144c0702()
209 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anon1ce6144c0a02()
210 std::vector<nnvm::NodeEntry> ret; in __anon1ce6144c0a02()
219 p = nnvm::Node::Create(); in __anon1ce6144c0a02()
[all …]
H A Dnp_moments_op.cc37 inline bool NumpyMomentsShape(const nnvm::NodeAttrs& attrs, in NumpyMomentsShape()
53 inline bool NumpyMomentsType(const nnvm::NodeAttrs& attrs, in NumpyMomentsType()
77 .set_attr<nnvm::FListInputNames>("FListInputNames",
81 .set_attr<nnvm::FListOutputNames>("FListOutputNames",
85 .set_attr<nnvm::FNumVisibleOutputs>("FNumVisibleOutputs",
105 .set_attr<nnvm::FListInputNames>("FListInputNames",
109 .set_attr<nnvm::FListOutputNames>("FListOutputNames",
113 .set_attr<nnvm::FNumVisibleOutputs>("FNumVisibleOutputs",
153 .set_attr<nnvm::FNumVisibleOutputs>("FNumVisibleOutputs",
161 .set_attr<nnvm::FListInputNames>("FListInputNames",
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_elemwise_unary_op_basic.cc52 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
57 .set_attr<nnvm::FInplaceIdentity>("FInplaceIdentity",
62 .set_attr<nnvm::FListInputNames>("FListInputNames",
183 .set_attr<nnvm::FListInputNames>("FListInputNames",
252 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
256 .set_attr<nnvm::FListInputNames>("FListInputNames",
446 .set_attr<nnvm::FListInputNames>("FListInputNames",
453 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
468 .set_attr<nnvm::FListInputNames>("FListInputNames",
476 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
[all …]
H A Dnp_elemwise_broadcast_op_extended.cc54 .set_attr<nnvm::TIsBackward>("TIsBackward", true)
55 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
69 .set_attr<nnvm::FListInputNames>("FListInputNames",
75 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
90 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
108 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
123 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
141 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
159 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
174 .set_attr<nnvm::FInplaceOption>("FInplaceOption",
[all …]
H A Dnp_where_op.cc104 .set_attr<nnvm::FGradient>("FGradient",
108 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonc102be760302()
109 std::vector<nnvm::NodeEntry> ret; in __anonc102be760302()
118 p = nnvm::Node::Create(); in __anonc102be760302()
164 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonc102be760702()
165 std::vector<nnvm::NodeEntry> ret; in __anonc102be760702()
174 p = nnvm::Node::Create(); in __anonc102be760702()
209 [](const nnvm::ObjectPtr& n, const std::vector<nnvm::NodeEntry>& ograds) { in __anonc102be760a02()
210 std::vector<nnvm::NodeEntry> ret; in __anonc102be760a02()
219 p = nnvm::Node::Create(); in __anonc102be760a02()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dnode_op_util.h36 const nnvm::ObjectPtr &dependent_node;
41 nnvm::NodeEntry mul(const nnvm::NodeEntry &lhs, const nnvm::NodeEntry &rhs) { in mul()
47 nnvm::NodeEntry mul(const nnvm::NodeEntry &x, double scalar) { in mul()
55 nnvm::NodeEntry mul(double scalar, const nnvm::NodeEntry &x) { in mul()
59 nnvm::NodeEntry div(const nnvm::NodeEntry &lhs, const nnvm::NodeEntry &rhs) { in div()
65 nnvm::NodeEntry square(const nnvm::NodeEntry &x) { in square()
66 return nnvm::NodeEntry{mxnet::op::MakeNode("square", in square()
71 nnvm::NodeEntry reciprocal(const nnvm::NodeEntry &x) { in reciprocal()
77 nnvm::NodeEntry negative(const nnvm::NodeEntry &x) { in negative()
78 return nnvm::NodeEntry{mxnet::op::MakeNode("negative", in negative()
[all …]
H A Dgraph_editor.cc30 namespace nnvm { namespace
40 std::vector<nnvm::Symbol *> GetInputSymbols(const nnvm::Symbol &sym) { in GetInputSymbols()
41 nnvm::Graph g; in GetInputSymbols()
42 std::vector<nnvm::Symbol *> input_syms; in GetInputSymbols()
47 const nnvm::Node &n = *idx[i].source; in GetInputSymbols()
51 nnvm::Symbol *s = new nnvm::Symbol(); in GetInputSymbols()
69 nnvm::NodeEntry e; in CutGraphInputs()
76 std::vector<nnvm::ObjectPtr> var_nodes; in CutGraphInputs()
88 nnvm::ObjectPtr n; in CutGraphInputs()
92 nnvm::Symbol sym; in CutGraphInputs()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dnode_op_util.h36 const nnvm::ObjectPtr &dependent_node;
41 nnvm::NodeEntry mul(const nnvm::NodeEntry &lhs, const nnvm::NodeEntry &rhs) { in mul()
47 nnvm::NodeEntry mul(const nnvm::NodeEntry &x, double scalar) { in mul()
55 nnvm::NodeEntry mul(double scalar, const nnvm::NodeEntry &x) { in mul()
59 nnvm::NodeEntry div(const nnvm::NodeEntry &lhs, const nnvm::NodeEntry &rhs) { in div()
65 nnvm::NodeEntry square(const nnvm::NodeEntry &x) { in square()
66 return nnvm::NodeEntry{mxnet::op::MakeNode("square", in square()
71 nnvm::NodeEntry reciprocal(const nnvm::NodeEntry &x) { in reciprocal()
77 nnvm::NodeEntry negative(const nnvm::NodeEntry &x) { in negative()
78 return nnvm::NodeEntry{mxnet::op::MakeNode("negative", in negative()
[all …]
H A Dgraph_editor.cc30 namespace nnvm { namespace
40 std::vector<nnvm::Symbol *> GetInputSymbols(const nnvm::Symbol &sym) { in GetInputSymbols()
41 nnvm::Graph g; in GetInputSymbols()
42 std::vector<nnvm::Symbol *> input_syms; in GetInputSymbols()
47 const nnvm::Node &n = *idx[i].source; in GetInputSymbols()
51 nnvm::Symbol *s = new nnvm::Symbol(); in GetInputSymbols()
69 nnvm::NodeEntry e; in CutGraphInputs()
76 std::vector<nnvm::ObjectPtr> var_nodes; in CutGraphInputs()
88 nnvm::ObjectPtr n; in CutGraphInputs()
92 nnvm::Symbol sym; in CutGraphInputs()
[all …]

12345678910>>...56