D | scatter_nd_op_cpu_impl.h | 47 template <typename T, typename Input, typename Update, typename Output, 52 Update update, Output output); 55 template <typename T, typename Input, typename Update, typename Output> 56 class UpdateExecutor<T, Input, Update, Output, 60 Update update, Output output) { in Execute() argument 61 output.device(device) = update; in Execute() 65 template <typename T, typename Input, typename Update, typename Output> 66 class UpdateExecutor<T, Input, Update, Output, scatter_nd_op::UpdateOp::ADD> { 69 Update update, Output output) { in Execute() argument 70 output.device(device) += update; in Execute() [all …]
|