Searched defs:ScatterNdFunctor (Results 1 – 4 of 4) sorted by relevance
32 struct ScatterNdFunctor { struct
125 struct ScatterNdFunctor<CPUDevice, T, Index, NDIMS, scatter_op::UpdateOp::ADD> { struct126 Index operator()(const CPUDevice &d, in operator ()()
49 struct ScatterNdFunctor { struct
112 struct ScatterNdFunctor<GPUDevice, T, Index, op, IXDIM> { struct113 Index operator()( in operator ()()