HomeSort by relevance Sort by last modified time
    Searched refs:REGISTER_SCATTER_KERNEL (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
scatter_op.cc 220 #define REGISTER_SCATTER_KERNEL(type, dev, name, op) \
225 REGISTER_SCATTER_KERNEL(type, dev, "ScatterAdd", scatter_op::UpdateOp::ADD); \
226 REGISTER_SCATTER_KERNEL(type, dev, "ScatterDiv", scatter_op::UpdateOp::DIV); \
227 REGISTER_SCATTER_KERNEL(type, dev, "ScatterMul", scatter_op::UpdateOp::MUL); \
228 REGISTER_SCATTER_KERNEL(type, dev, "ScatterSub", scatter_op::UpdateOp::SUB);
231 REGISTER_SCATTER_KERNEL(type, dev, "ScatterUpdate", \
275 #undef REGISTER_SCATTER_KERNEL
resource_variable_ops.cc 623 #define REGISTER_SCATTER_KERNEL(type, dev, name, op) \
629 REGISTER_SCATTER_KERNEL(type, dev, "ResourceScatterAdd", \
631 REGISTER_SCATTER_KERNEL(type, dev, "ResourceScatterUpdate", \
640 REGISTER_SCATTER_KERNEL(string, CPU, "ResourceScatterUpdate",
656 #undef REGISTER_SCATTER_KERNEL

Completed in 69 milliseconds