Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DIndexing.cu1554 Tensor & masked_fill__cuda(Tensor& self, const Tensor & mask, const Scalar& value) { in masked_fill__cuda() function
1584 Tensor & masked_fill__cuda(Tensor& self, const Tensor & mask, const Tensor & value) { in masked_fill__cuda() function