/dports/misc/tvm/incubator-tvm-0.6.1/docs/ |
H A D | nnvm_top.rst | 48 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 D | nnvm_top.rst | 48 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 D | elemwise_unary_op_trig.cc | 52 [](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 D | la_op.cc | 113 .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 D | elemwise_unary_op_pow.cc | 50 .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 D | elemwise_unary_op_basic.cc | 90 [](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 D | elemwise_unary_op_logexp.cc | 111 .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 D | matrix_op.cc | 43 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 D | elemwise_unary_op_trig.cc | 52 [](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 D | la_op.cc | 113 .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 D | elemwise_unary_op_pow.cc | 50 .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 D | elemwise_unary_op_basic.cc | 90 [](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 D | elemwise_unary_op_logexp.cc | 111 .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 D | matrix_op.cc | 43 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 D | np_elemwise_unary_op_basic.cc | 52 .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 D | np_elemwise_broadcast_op_extended.cc | 54 .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 D | np_where_op.cc | 104 .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 D | np_moments_op.cc | 37 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 D | np_elemwise_unary_op_basic.cc | 52 .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 D | np_elemwise_broadcast_op_extended.cc | 54 .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 D | np_where_op.cc | 104 .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 D | node_op_util.h | 36 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 D | graph_editor.cc | 30 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 D | node_op_util.h | 36 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 D | graph_editor.cc | 30 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 …]
|