| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | roi_align_grad.cc | 109 void ROIAlignGrad::set_sample_num(const int64_t sample_num) { in set_sample_num()
116 const int64_t sample_num) { in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/session/ |
| D | dynamic_session.h | 43 void SetSampleNum(int sample_num) { sample_num_ = sample_num; } in SetSampleNum()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | roi_align.cc | 140 void ROIAlign::set_sample_num(const int64_t sample_num) { (void)this->AddAttr(kSampleNum, api::Make… in set_sample_num()
157 const int64_t sample_num, const int64_t roi_end_mode) { in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | roi_align_impl.cu | 70 …_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, const int sample_num, in bin_box()
120 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignKernel()
176 … const int sample_num, int roi_end_mode, const int channels, const int height, const int width, in ROIAlign()
208 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignGradKernel()
270 … const T spatial_scale, const int sample_num, int roi_end_mode, const int channels, in ROIAlignGrad()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_roi_align.py | 23 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num, roi_end_mode): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | parameterized_truncated_normal_cpu_kernel.cc | 174 int sample_num = 0; in GenerateCase1() local
205 int sample_num = 0; in GenerateCase2() local
242 int sample_num = 0; in GenerateCase3() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_roi_align_grad_op.py | 26 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num, is_dyn_rank=False): argument
|
| D | test_roi_align_op.py | 26 …def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num, roi_end_mode, is_dyn_ra… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/debugger/ |
| D | debugger_utils.cc | 101 const uint32_t sample_mode, const uint32_t sample_num, const bool async_copy) { in LoadInputs()
157 const uint32_t sample_mode, const uint32_t sample_num) { in LoadOutputs()
275 uint32_t sample_num = GetSampleNum(); in ReadDataAndDump() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | roialign_tensorrt.h | 42 … int sample_num, int roi_end_mode, int channel, int height, int width, int roi_rows, in ROIAlignPlugin()
|
| D | roialign_tensorrt.cc | 50 int sample_num = op->get_sample_num(); in AddInnerOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/debug/data_dump/ |
| D | dump_json_parser.h | 77 uint32_t sample_num() const { return sample_num_; } in sample_num() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/ |
| D | gptq_quantizer.cc | 244 int sample_num) -> bool { in RunKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | parameterized_truncated_normal.cc | 94 int sample_num = 0; in Generate() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | roi_align_cpu_kernel.cc | 75 …_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, const int sample_num, in bin_box()
|
| D | roi_align_grad_cpu_kernel.cc | 79 …_box(int thread_idx, const T *roi_boxes, int roi_cols, const T spatial_scale, const int sample_num, in bin_box()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/ |
| D | dump_json_parser.cc | 533 auto sample_num = CheckJsonKeyExist(*e2e_dump_setting, kSampleNum); in ParseE2eDumpSetting() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | nn_ops.py | 3585 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num=2, roi_end_mode=1): argument
|
| D | _grad_ops.py | 1584 def __init__(self, pooled_height, pooled_width, spatial_scale, sample_num=2): argument
|
| /third_party/mesa3d/src/compiler/glsl/ |
| D | builtin_functions.cpp | 8554 ir_variable *sample_num = in_var(&glsl_type_builtin_int, "sample_num"); in _interpolateAtSample() local
|