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

  /external/tensorflow/tensorflow/core/kernels/
scatter_op.cc 213 #define REGISTER_SCATTER_KERNEL_INDEX(type, index_type, dev, name, op) \
221 REGISTER_SCATTER_KERNEL_INDEX(type, int32, dev, name, op); \
222 REGISTER_SCATTER_KERNEL_INDEX(type, int64, dev, name, op);
276 #undef REGISTER_SCATTER_KERNEL_INDEX
resource_variable_ops.cc 614 #define REGISTER_SCATTER_KERNEL_INDEX(type, index_type, dev, name, op) \
624 REGISTER_SCATTER_KERNEL_INDEX(type, int32, dev, name, op); \
625 REGISTER_SCATTER_KERNEL_INDEX(type, int64, dev, name, op);
657 #undef REGISTER_SCATTER_KERNEL_INDEX

Completed in 247 milliseconds