Searched defs:per_sample_weights_maybe_owned (Results 1 – 2 of 2) sorted by relevance
315 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in _embedding_bag_forward_only_cuda() local349 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in _embedding_bag_cuda() local435 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in _embedding_bag_dense_backward_cuda() local
1232 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in embedding_bag() local1275 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in _embedding_bag_forward_only_cpu() local1298 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in _embedding_bag_cpu() local1399 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in _embedding_bag_backward_symint() local1756 …c10::MaybeOwned<Tensor> per_sample_weights_maybe_owned = at::borrow_from_optional_tensor(per_sampl… in _embedding_bag_sparse_backward_symint() local