Searched defs:conv2d (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | conv2d_unify_mindir.cc | 45 bool NeedUpdate(const CNodePtr &conv2d, std::vector<size_t> in_shape, std::vector<size_t> out_shape… in NeedUpdate()
93 CNodePtr CreateTranspose(const FuncGraphPtr &graph, const CNodePtr &conv2d, const AnfNodePtr &input… in CreateTranspose()
136 CNodePtr CreateDepthwiseConv2D(const FuncGraphPtr &graph, const CNodePtr &conv2d, const CNodePtr &t… in CreateDepthwiseConv2D()
212 void SetCommonAttrs(const CNodePtr &conv2d, const CNodePtr &depth_conv) { in SetCommonAttrs()
223 void SetConv2DAttrs(const CNodePtr &conv2d, const CNodePtr &depth_conv) { in SetConv2DAttrs()
270 auto conv2d = node->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pynative/ |
| D | ops_test.py | 54 def conv2d(x, weight, bias=None, stride=1, pad=0, function
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pipeline/infer/ |
| D | primitive_test.py | 68 conv2d = P.Conv2D(64, variable
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | graph_runner_test.cc | 58 PrimitivePtr conv2d = prim::kPrimConv2D; in MakeGeGraph() local
|
| D | convert_test.cc | 91 PrimitivePtr conv2d = prim::kPrimConv2D; in TEST_F() local
|
| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 328 auto conv2d = tuple->cast<CNodePtr>()->input(1); in TEST_F() local
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_me.py | 231 def conv2d(x, weight, bias=None, stride=1, pad=0, function
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | conv2d_info.cc | 762 AnfNodePtr conv2d; in ComputeReplaceGraph() local
|