Is there a way to utilize ATen full lib by calling caffe2.python.core.CreateOperators?

Hi I just wondering if there is a way to utilize ATen full lib by creating like

op = core.CreateOperator(
    ["X2", "X1"],

I wanted to use grid_sampler function in ATen lib and I have checked the native_functions.yaml in aten/src/ATen/native which has function grid_sampler.

However when I run operators above, it shows an error claimed that

Attempting to run unknown ATen operator configuration: grid_sampler

It’s really appreciated that if someone can answer my question!