GridSampler3DFusionPass

Description

Replaces the GridSampler3D operator with the GridSample operator.

• Scenario 1: The input data format is NCDHW or ND.

  Before: After:

• Scenario 2: The input data format is NDHWC.

  Before: After:

Restrictions

• The dynamic shape scenario is not supported.
• The x and y data formats must be NCDHW, NDHWC, or ND and must be the same, and the shape must be 5D.
• The shape of grid must be 5D and the last dimension must be 3.
• Data type restrictions:
  • Atlas A2 training product/Atlas A2 inference product: The data type can be FLOAT32 or FLOAT16.

Availability

Atlas A2 training product/Atlas A2 inference product