Home
last modified time | relevance | path

Searched refs:finfer_shape (Results 1 – 14 of 14) sorted by relevance

/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/pass/
H A Dinfer_shape_type.cc44 static auto& finfer_shape = in InferAttr() local
167 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()
H A Dgradient.cc168 static auto& finfer_shape = Op::GetAttr<FInferShape>("FInferShape"); in Gradient() local
221 if (finfer_shape.contains(git->node->op())) { in Gradient()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/nnvm/src/pass/
H A Dinfer_shape_type.cc38 static auto& finfer_shape = Op::GetAttr<FInferNodeEntryAttr<AttrType>>(infer_name); in InferAttr() local
157 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()
H A Dgradient.cc163 static auto& finfer_shape = Op::GetAttr<FInferShape>("FInferShape"); in Gradient() local
215 if (finfer_shape.contains(git->node->op())) { in Gradient()
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/pass/
H A Dinfer_shape_type.cc44 static auto& finfer_shape = in InferAttr() local
167 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()
H A Dgradient.cc168 static auto& finfer_shape = Op::GetAttr<FInferShape>("FInferShape"); in Gradient() local
221 if (finfer_shape.contains(git->node->op())) { in Gradient()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dgradient.cc169 static auto& finfer_shape = Op::GetAttr<mxnet::FInferShape>("FInferShape"); in Gradient() local
222 if (finfer_shape.contains(git->node->op())) { in Gradient()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dgradient.cc169 static auto& finfer_shape = Op::GetAttr<mxnet::FInferShape>("FInferShape"); in Gradient() local
222 if (finfer_shape.contains(git->node->op())) { in Gradient()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/executor/
H A Dinfer_graph_attr_pass.cc289 static auto& finfer_shape = in InferAttr() local
421 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()
574 static auto& finfer_shape = in InferShapeAttr() local
724 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferShapeAttr()
H A Dgraph_executor.cc1431 static auto& finfer_shape = nnvm::Op::GetAttr<mxnet::FInferShape>("FInferShape"); in RunOps() local
1490 if (finfer_shape.count(op)) { in RunOps()
1499 auto finfer = finfer_shape[op]; in RunOps()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/executor/
H A Dinfer_graph_attr_pass.cc289 static auto& finfer_shape = in InferAttr() local
421 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferAttr()
574 static auto& finfer_shape = in InferShapeAttr() local
724 auto finfer = finfer_shape.get(inode.source->op(), fdefault); in InferShapeAttr()
H A Dgraph_executor.cc1431 static auto& finfer_shape = nnvm::Op::GetAttr<mxnet::FInferShape>("FInferShape"); in RunOps() local
1490 if (finfer_shape.count(op)) { in RunOps()
1499 auto finfer = finfer_shape[op]; in RunOps()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/cpp/include/
H A Dtest_core_op.h470 static auto& finfer_shape = Op::GetAttr<mxnet::FInferShape>("FInferShape"); variable
471 if (finfer_shape.count(op_)) {
472 mxnet::FInferShape call_infer_shapes = finfer_shape[op_];
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/cpp/include/
H A Dtest_core_op.h470 static auto& finfer_shape = Op::GetAttr<mxnet::FInferShape>("FInferShape"); variable
471 if (finfer_shape.count(op_)) {
472 mxnet::FInferShape call_infer_shapes = finfer_shape[op_];