Searched defs:output_mask (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | rcwm_small_impl.cu | 43 __global__ void L2Rcwm(int seedc, int input_size, const K *input, K *output_mask, S *output_index, … in L2Rcwm()
126 void RCWMScaleK(int seedc, int input_size, K *input, int k, S *output_index, K *output_mask, cudaSt… in RCWMScaleK()
146 …hoiceWithMaskSmall(int input_size, int seedc, int count, K *input, S *output_index, K *output_mask, in CalRandomChoiceWithMaskSmall()
|
| D | random_choice_with_mask_impl.cu | 185 T *output_mask, S *index_buff, S *rank_buff, S *Tnum_buff) { in MoveToOutput()
231 … const T *input, S *output_index, T *output_mask, S *index_buff, S *mask_buff, S *rank_buff, in CalRandomChoiceWithMask()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/random/ |
| D | random_choice_with_mask_gpu_kernel.h | 47 T *output_mask = GetDeviceAddress<T>(outputs, 1); in Launch() local
|
| /third_party/ffmpeg/libavcodec/ |
| D | dca_xll.h | 140 int output_mask; member
|
| /third_party/mesa3d/src/gallium/auxiliary/util/ |
| D | u_simple_shaders.c | 532 const char *output_mask, in util_make_fs_blit_msaa_gen()
|
| /third_party/weston/include/libweston/ |
| D | libweston.h | 1316 uint32_t output_mask; member
1437 uint32_t output_mask; member
|
| /third_party/mesa3d/src/amd/vulkan/ |
| D | radv_nir_to_llvm.c | 69 uint64_t output_mask; member
|