Home
last modified time | relevance | path

Searched defs:row_mask (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dnms_with_mask_impl.cu40 __global__ void MaskInit(int numSq, bool *row_mask) { in MaskInit()
111 bool *row_mask) { in NmsPass()
126 __global__ void ReducePass(const int num, bool *sel_boxes, bool *row_mask) { in ReducePass()
175 bool *row_mask, cudaStream_t cuda_stream) { in CalPreprocess()
192 …Nms(const int num, const float IOU_value, T *output, bool *sel_boxes, int box_size, bool *row_mask, in CalNms()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dnms_with_mask_cpu_kernel.cc76 void NMSWithMaskCPUKernel<T>::MaskInit(size_t numSq, bool *row_mask) { in MaskInit()
161 bool *row_mask) { in NmsPass()
178 void NMSWithMaskCPUKernel<T>::ReducePass(const int num, bool *sel_boxes, const bool *row_mask) { in ReducePass()
228 auto row_mask = reinterpret_cast<bool *>(workspace[ROW_MASK]->addr); in Launch() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dnms_with_mask_gpu_kernel.h55 bool *row_mask = GetDeviceAddress<bool>(workspace, 2); in Launch() local
/third_party/mesa3d/src/amd/compiler/
Daco_lower_to_hw_instr.cpp200 PhysReg vtmp_reg, ReduceOp op, unsigned dpp_ctrl, unsigned row_mask, in emit_int64_dpp_op()
387 ReduceOp op, unsigned size, unsigned dpp_ctrl, unsigned row_mask, unsigned bank_mask, in emit_dpp_op()
455 unsigned row_mask, unsigned bank_mask, bool bound_ctrl) in emit_dpp_mov()
Daco_ir.h1412 uint8_t row_mask : 4; member
/third_party/flutter/skia/third_party/externals/libpng/
Dpngrutil.c3365 static const png_uint_32 row_mask[2/*PACKSWAP*/][3/*depth*/][6] = in png_combine_row() local
/third_party/skia/third_party/externals/libpng/
Dpngrutil.c3365 static const png_uint_32 row_mask[2/*PACKSWAP*/][3/*depth*/][6] = in png_combine_row() local
/third_party/libpng/
Dpngrutil.c3365 static const png_uint_32 row_mask[2/*PACKSWAP*/][3/*depth*/][6] = in png_combine_row() local
/third_party/mesa3d/src/amd/llvm/
Dac_llvm_build.c3519 enum dpp_ctrl dpp_ctrl, unsigned row_mask, unsigned bank_mask, in _ac_build_dpp()
3539 enum dpp_ctrl dpp_ctrl, unsigned row_mask, unsigned bank_mask, in ac_build_dpp()