oneflow/python/nn/modules/dataset.py

oneflow/python/ops/user_data_ops.py

oneflow/user/kernels/image_batch_align_kernel.cpp

oneflow/user/ops/image_batch_align_op.cpp