Searched defs:masked_grad (Results 1 – 2 of 2) sorted by relevance
555 const bool masked_grad = masked_grad_.value_or(true); in to_dense_backward() local601 …ense(const Tensor& tensor, std::optional<c10::ScalarType> dtype, std::optional<bool> masked_grad) { in to_dense()634 std::optional<bool> masked_grad) { in sparse_compressed_to_dense()
29 …se(const Tensor& mkldnn_tensor, std::optional<ScalarType> dtype, std::optional<bool> masked_grad) { in mkldnn_to_dense()