Home
last modified time | relevance | path

Searched defs:get_kernel_size (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/core/ops/grad/
Dpool_grad.cc77 std::vector<int64_t> PoolGrad::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::PoolGrad
Dde_conv2d_grad_filter.cc66 std::vector<int64_t> DeConv2DGradFilter::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::DeConv2DGradFilter
Dgroup_conv2d_grad_input.cc67 std::vector<int64_t> GroupConv2DGradInput::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::GroupConv2DGradInput
Dconv2d_backprop_filter.cc156 std::vector<int64_t> Conv2DBackpropFilter::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::Conv2DBackpropFilter
Dconv2d_backprop_input.cc248 std::vector<int64_t> Conv2DBackpropInput::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::Conv2DBackpropInput
/third_party/mindspore/mindspore/core/ops/
Dadder.cc54 std::vector<int64_t> Adder::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::Adder
Davg_pool.cc40 std::vector<int64_t> AvgPool::get_kernel_size() const { return GetValue<std::vector<int64_t>>(GetAt… in get_kernel_size() function in mindspore::ops::AvgPool
Dconv2d_transpose.cc128 std::vector<int64_t> Conv2DTranspose::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::Conv2DTranspose
Dmax_pool.cc40 std::vector<int64_t> MaxPool::get_kernel_size() const { return GetValue<std::vector<int64_t>>(GetAt… in get_kernel_size() function in mindspore::ops::MaxPool
Dconv2d.cc318 std::vector<int64_t> Conv2D::get_kernel_size() const { in get_kernel_size() function in mindspore::ops::Conv2D