Searched defs:image_size_h (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | prior_box.cc | 67 void PriorBox::set_image_size_h(const int64_t image_size_h) { in set_image_size_h()
110 … const int64_t image_size_w, const int64_t image_size_h, const float step_w, const float step_h, in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | affine_grid_gpu_kernel.cc | 41 const std::vector<int64_t> &image_size_h, const std::string &kernel_name) { in PreLaunchKernel4D()
61 const std::vector<int64_t> &image_size_h, const std::string &kernel_name) { in PreLaunchKernel5D()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/ |
| D | prior_box_parameter.h | 31 int32_t image_size_h; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_1.cc | 716 …atios, const std::vector<float> &variances, int64_t image_size_w, int64_t image_size_h, float step… in MindIR_PriorBox_CreatePrimitive()
923 void MindIR_PriorBox_SetImageSizeH(PrimitivePtr *primitive, int64_t image_size_h) { in MindIR_PriorBox_SetImageSizeH()
|