Searched refs:getSparseGradIndices (Results 1 – 2 of 2) sorted by relevance
51 if (bucket.getSparseGradIndices().has_value()) { in runHook()53 opts.sparseIndices = bucket.getSparseGradIndices().value(); in runHook()
75 std::optional<at::Tensor>& getSparseGradIndices() { in getSparseGradIndices() function in c10d::GradBucket