diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/Gather.cc')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/Gather.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Gather.cc b/runtimes/pure_arm_compute/src/internal/op/Gather.cc index 6c0dbaf75..bc517d28c 100644 --- a/runtimes/pure_arm_compute/src/internal/op/Gather.cc +++ b/runtimes/pure_arm_compute/src/internal/op/Gather.cc @@ -53,11 +53,11 @@ Param::Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, // Each input should be interpreted as follows: // - // 0 -> LHS Tensor Index - // 1 -> RHS Tensor Index + // 0 -> input Tensor Index + // 1 -> indices Tensor Index // 2 -> axis Index - lhs_index = inputs[0]; - rhs_index = inputs[1]; + ifm_index = inputs[0]; + indices_index = inputs[1]; axis_index = inputs[2]; } |