Migrate dirichlet from CUDA_tensor_apply3 to TensorIterator (#34021)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34021
Test Plan: Imported from OSS
Differential Revision: D20196082
Pulled By: VitalyFedyunin
fbshipit-source-id: 9736a0ebbc529975e95a4f996dbc28e070cf1e63