Searched defs:NodePtrList (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | op_node.h | 63 virtual void Check(const NodePtrList &, const DAttrs &) {} in Check()
96 DFormat InferFormat(const NodePtrList &, const DAttrs &attrs) override { in InferFormat()
130 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; }; in InferFormat()
196 DFormat InferFormat(const NodePtrList &, const DAttrs &attrs) override { in InferFormat()
207 std::vector<DShape> InferShape(const NodePtrList &, const DAttrs &attrs) override { in InferShape()
214 …std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &) override { return {TypeId::kNum… in InferType()
227 …std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &) override { return {TypeId::kNum… in InferType()
228 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; }; in InferFormat()
240 std::vector<TypeId> InferType(const NodePtrList &, const DAttrs &attrs) override { in InferType()
243 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; } in InferFormat()
[all …]
|
| D | node.h | 58 using NodePtrList = std::vector<NodePtr>; variable
|
| D | op_node.cc | 479 std::vector<TypeId> ArgReduceOp::InferType(const NodePtrList &, const DAttrs &attrs) { in InferType()
1096 std::vector<DShape> StandardNormalOp::InferShape(const NodePtrList &, const DAttrs &attrs) { in InferShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expander/base/ |
| D | node.h | 47 using NodePtrList = std::vector<NodePtr>; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expander/mindir_adapter/ |
| D | infer.h | 31 virtual void HandleInputs(const NodePtrList &) {} in HandleInputs()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/expanders/ |
| D | constant_of_shape.cc | 31 NodePtrList Expand(const NodePtrList &) override { in Expand()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/ |
| D | func_builder.h | 30 using NodePtrList = expander::NodePtrList; variable
|
| D | func_pass.h | 35 using NodePtrList = expander::NodePtrList; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/expander/core/ |
| D | node.h | 74 using NodePtrList = std::vector<NodePtr>; variable
|