Searched defs:mask_arg (Results 1 – 1 of 1) sorted by relevance
16 const Tensor& mask_arg, in masked_fill_scalar()125 const Tensor& mask_arg, in masked_fill_tensor()