Searched refs:scatter_nd (Results 1 – 4 of 4) sorted by relevance
65 scatter_nd = P.ScatterNd()72 scattered_out_indicators = scatter_nd(indices, out_indicators, shape(x))
134 scatter_nd = P.ScatterNd() variable
104 from .scatter_nd import _scatter_nd_tbe
131 scatter_nd = P.ScatterNd()159 sp_tensor = scatter_nd(idx_tensor, fill(dtype(dout), (out_indices_num,), 1), sp_shape)