Home
last modified time | relevance | path

Searched defs:n_input_plane (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dim2col_and_col2im_ext.cc141 void Col2ImCheckNInputPlane(const PrimitivePtr &primitive, int64_t n_input_plane, in Col2ImCheckNInputPlane()
319 auto n_input_plane = input_shape[input_shape.size() - kIndex2]; in REGISTER_SIMPLE_INFER() local
354 auto n_input_plane = input_shape[input_shape.size() - kIndex2]; in CheckValidation() local
424 auto n_input_plane = input_shape[input_rank - kIndex2]; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcol2im_cpu_kernel.cc94 const int64_t n_input_plane = x_shape_[kIndex1]; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dcol2im.cc74 const int64_t n_input_plane = x_shape[kInputIndex2]; in Col2ImShapeCheck() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dcol2im.cc225 const int64_t n_input_plane = input_shape[kIndex1]; in Col2imCompute() local