Searched defs:Conv2D (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/include/api/ops/ |
| D | ops.h | 28 struct MS_API Conv2D : public OpCell<Conv2D> { struct
29 Conv2D() : OpCell("Conv2D") {} in Conv2D() function
38 int out_channel;
39 std::vector<int> kernel_size;
40 int mode = 1;
41 std::string pad_mode = "valid";
42 std::vector<int> pad = {0, 0, 0, 0};
43 std::vector<int> stride = {1, 1, 1, 1};
44 std::vector<int> dilation = {1, 1, 1, 1};
45 int group = 1;
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | conv2d.h | 35 Conv2D() : PrimitiveC(kNameConv2D) { InitIOName({"x", "w"}, {"output"}); } in Conv2D() function
36 …explicit Conv2D(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x", "w"}, {"output"}… in Conv2D() function
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/ops/ |
| D | ops.cc | 19 Conv2D::Conv2D(int out_channel, const std::vector<int> &kernel_size, int mode, const std::string &p… in Conv2D() function in mindspore::Conv2D
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | conv2d.py | 35 class Conv2D(Expander): class
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/tbe/ |
| D | tbe_json_creator_test.py | 33 Conv2D = P.Conv2D(out_channel=32, kernel_size=3) variable
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | op_infer.py | 366 class Conv2D(OpInfer): class
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 256 MS_TRT_CONVERTER_FUNC_REG(Conv2D) { in MS_TRT_CONVERTER_FUNC_REG() argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/ |
| D | image_process.cc | 327 bool Conv2D(const LiteMat &src, const LiteMat &kernel, LiteMat &dst, LDataType dst_type, PaddBorder… in Conv2D() function
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 1292 class Conv2D(Primitive): class
|