[wasm] Add Gather, GatherNd kernels.

[annxingyuan]

To see the logs from the Cloud Build CI, please join either our discussion or announcement mailing list.

---

This change is 

[nsthorat]

Reviewed 12 of 12 files at r1.
Reviewable status: complete! 1 of 1 approvals obtained (waiting on @annxingyuan, @dsmilkov, and @nsthorat)

---

tfjs-backend-wasm/src/cc/kernels/Gather.cc, line 50 at r1 (raw file):

#endif

void Gather(const size_t x_id, const DType dtype, const size_t* x_strides_ptr,

this should be int32_t

---

tfjs-backend-wasm/src/cc/kernels/Gather.cc, line 52 at r1 (raw file):

void Gather(const size_t x_id, const DType dtype, const size_t* x_strides_ptr,
            const size_t strides_size, const size_t indices_id,
            const size_t axis, const size_t* out_strides_ptr,

this should be int32_t

---

tfjs-backend-wasm/src/cc/kernels/GatherNd.cc, line 25 at r1 (raw file):

template <typename T>
void gathernd_impl(const T* x_ptr, const int* indices_ptr,

this should be int32_t

---

tfjs-backend-wasm/src/cc/kernels/GatherNd.cc, line 28 at r1 (raw file):

                   const size_t num_slices, const size_t slice_rank,
                   const size_t slice_size,
                   const std::vector<size_t>& strides_ptr, T* out_buf_ptr) {

this should be int32_t

---

tfjs-backend-wasm/src/cc/kernels/GatherNd.cc, line 60 at r1 (raw file):

void GatherNd(const size_t x_id, const DType dtype, const size_t indices_id,
              const size_t num_slices, const size_t slice_rank,
              const size_t slice_size, const size_t* strides_ptr,

this should be int32_t

[annxingyuan]

Reviewable status: complete! 1 of 1 approvals obtained (waiting on @dsmilkov)

---

tfjs-backend-wasm/src/cc/kernels/Gather.cc, line 50 at r1 (raw file):

Previously, nsthorat (Nikhil Thorat) wrote…

this should be int32_t

Done

---

tfjs-backend-wasm/src/cc/kernels/Gather.cc, line 52 at r1 (raw file):

Previously, nsthorat (Nikhil Thorat) wrote…

this should be int32_t

Done

---

tfjs-backend-wasm/src/cc/kernels/GatherNd.cc, line 25 at r1 (raw file):

Previously, nsthorat (Nikhil Thorat) wrote…

this should be int32_t

Done

---

tfjs-backend-wasm/src/cc/kernels/GatherNd.cc, line 28 at r1 (raw file):

Previously, nsthorat (Nikhil Thorat) wrote…

this should be int32_t

Done

---

tfjs-backend-wasm/src/cc/kernels/GatherNd.cc, line 60 at r1 (raw file):

Previously, nsthorat (Nikhil Thorat) wrote…

this should be int32_t

Done