Searched defs:ScatterNd (Results 1 – 3 of 3) sorted by relevance
34 …ScatterNd() : PrimitiveC(kNameScatterNd) { InitIOName({"indices", "update", "shape"}, {"output"});… in ScatterNd() function
50 void ScatterNd(S *indices, T *update, T *output, const size_t &block_size, const size_t &input_size, in ScatterNd() function
3538 class ScatterNd(PrimitiveWithInfer): class