Searched defs:InferFormat (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | op_node.h | 66 …virtual DFormat InferFormat(const NodePtrList &inputs, const DAttrs &) { return inputs[0]->format;… in InferFormat() function
96 DFormat InferFormat(const NodePtrList &, const DAttrs &attrs) override { in InferFormat() function
130 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; }; in InferFormat() function
196 DFormat InferFormat(const NodePtrList &, const DAttrs &attrs) override { in InferFormat() function
228 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; }; in InferFormat() function
243 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; } in InferFormat() function
287 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; }; in InferFormat() function
300 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; }; in InferFormat() function
349 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; } in InferFormat() function
378 DFormat InferFormat(const NodePtrList &, const DAttrs &) override { return kOpFormat_DEFAULT; } in InferFormat() function
|
| D | op_node.cc | 440 DFormat ElemwiseOp::InferFormat(const NodePtrList &inputs, const DAttrs &) { in InferFormat() function in mindspore::graphkernel::inner::ElemwiseOp
484 DFormat TransposeOp::InferFormat(const NodePtrList &inputs, const DAttrs &attrs) { in InferFormat() function in mindspore::graphkernel::inner::TransposeOp
689 DFormat Conv2dOp::InferFormat(const NodePtrList &inputs, const DAttrs &attrs) { in InferFormat() function in mindspore::graphkernel::inner::Conv2dOp
|