Searched defs:clamped_indices (Results 1 – 1 of 1) sorted by relevance
683 torch::List<std::optional<Tensor>> clamped_indices(indices); in _unsafe_masked_index() local728 torch::List<std::optional<Tensor>> clamped_indices(indices); in _unsafe_masked_index_put_accumulate() local