Searched refs:kPrimConv2DBackpropFilter (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.h | 44 {prim::kPrimConv2DBackpropFilter->name(), {{0, 1}, {0}}},
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 31 Register(prim::kPrimConv2DBackpropFilter->name(), {2}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | bn_reduce_grad_conv2d_backprop_filter_fusion.cc | 68 VectorRef pattern({prim::kPrimConv2DBackpropFilter, bnreducegrad, X}); in DefinePattern()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | conv2d_backprop_filter.cc | 243 REGISTER_PRIMITIVE_EVAL_IMPL(Conv2DBackpropFilter, prim::kPrimConv2DBackpropFilter, Conv2DBackpropF…
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 54 const auto kConv2DBackpropFilter = prim::kPrimConv2DBackpropFilter->name(); in GetDependsFormMap()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nn_calculation_ops_declare.cc | 79 REG_ADPT_DESC(Conv2DBackpropFilterD, prim::kPrimConv2DBackpropFilter->name(), ADPT_DESC(Conv2DBackp…
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | conv2d_unify_mindir.cc | 319 VectorRef pattern({prim::kPrimConv2DBackpropFilter, dout, input, filter_size}); in DefinePattern()
|
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | ops_test.cc | 409 ASSERT_EQ(prim->name(), kPrimConv2DBackpropFilter->name()); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 202 auto prim = prim::kPrimConv2DBackpropFilter; in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 315 inline const PrimitivePtr kPrimConv2DBackpropFilter = std::make_shared<Primitive>("Conv2DBackpropFi… variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 148 {prim::kPrimConv2DBackpropFilter->name(), {{{0, 1}, {1, 2}, {2, 0}}, {{1, 0}, {2, 1}, {0, 2}}}}};
156 {prim::kPrimConv2DBackpropFilter->name(), {{{0, 1}, {1, 0}}, {{0, 1}, {1, 0}}}},
|