Searched defs:MaskedSelect (Results 1 – 3 of 3) sorted by relevance
36 MaskedSelect() : BaseOperator(kNameMaskedSelect) { InitIOName({"x", "mask"}, {"output"}); } in MaskedSelect() function
47 cudaError_t MaskedSelect(T *input_ptr, bool *mask_ptr, size_t *index_ptr, const std::vector<int64_t… in MaskedSelect() function
4359 class MaskedSelect(PrimitiveWithCheck): class