GridSampler3DFusionPass

融合模式

该融合规则主要是使用GridSample算子替换GridSampler3D算子。

• 场景一：输入的数据格式为NCDHW、ND。

  融合为

• 场景二：输入的数据格式为NDHWC。

  融合为

使用约束

• 不支持动态shape场景。
• 仅支持x、y数据格式为NCDHW、NDHWC、ND，shape为5维，且数据格式一致的场景。
• grid的shape必须为5维且最后一维值为3。
• 数据类型限制：
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件：数据类型支持FLOAT32、FLOAT16。

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件