Searched defs:ScatterNdFunctor (Results 1 – 4 of 4) sorted by relevance
32 struct ScatterNdFunctor { struct
155 struct ScatterNdFunctor<CPUDevice, T, Index, NDIMS, scatter_op::UpdateOp::ADD> { struct156 Index operator()(const CPUDevice &d, in operator ()()
48 struct ScatterNdFunctor { struct
128 struct ScatterNdFunctor<GPUDevice, T, Index, op, IXDIM> { struct129 Index operator()( in operator ()()