Home
last modified time | relevance | path

Searched defs:dilation_array (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dmax_pool_with_mask.cc132 const auto &dilation_array = dilation_array_opt.value(); in InferShape() local
184 const auto &dilation_array = dilation_array_opt.value(); in CheckValidation() local
Dmax_pool_with_indices.cc158 const auto &dilation_array = dilation_array_opt.value(); in InferShape() local
209 const auto &dilation_array = dilation_array_opt.value(); in CheckValidation() local
Dim2col_and_col2im_ext.cc189 auto dilation_array = GetArrayValue<int64_t>(input_values[kIndex2]).value(); in InferShape() local
417 auto dilation_array = GetArrayValue<int64_t>(input_values[kIndex3]).value(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dmax_pool_with_mask.cc38 auto dilation_array = ConvertValueTupleToVector<int64_t>(dilation); in MaxPoolWithMaskAscendCall() local
Dmax_pool_with_indices.cc38 auto dilation_array = ConvertValueTupleToVector<int64_t>(dilation); in MaxPoolWithIndicesAscendCall() local
Dmax_pool_grad_with_mask.cc38 auto dilation_array = ConvertValueTupleToVector<int64_t>(dilation); in MaxPoolGradWithMaskAscendCall() local
Dmax_pool_grad_with_indices.cc39 auto dilation_array = ConvertValueTupleToVector<int64_t>(dilation); in MaxPoolGradWithIndicesAscendCall() local