diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/Gather.h')
| -rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/Gather.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Gather.h b/runtimes/pure_arm_compute/src/internal/op/Gather.h index 4470236eb..d40794f99 100644 --- a/runtimes/pure_arm_compute/src/internal/op/Gather.h +++ b/runtimes/pure_arm_compute/src/internal/op/Gather.h @@ -43,9 +43,9 @@ struct Param { int32_t ofm_index; //!< index for output feature map - int32_t lhs_index; //!< index for lhs tensor - int32_t rhs_index; //!< index for rhs tensor - int32_t axis_index; //!< index for axis + int32_t ifm_index; //!< index for ifm tensor + int32_t indices_index; //!< index for indices tensor + int32_t axis_index; //!< index for axis /** * @brief Default Constructor |