Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:REGISTER_SCATTER_UPDATE (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/core/kernels/
scatter_op.cc	230 #define REGISTER_SCATTER_UPDATE(type, dev) \ 238 #define REGISTER_SCATTER_UPDATE_CPU(type) REGISTER_SCATTER_UPDATE(type, CPU); 248 #define REGISTER_SCATTER_UPDATE_GPU(type) REGISTER_SCATTER_UPDATE(type, GPU); 260 #define REGISTER_SCATTER_UPDATE_SYCL(type) REGISTER_SCATTER_UPDATE(type, SYCL); 272 #undef REGISTER_SCATTER_UPDATE
resource_variable_ops.cc	648 #define REGISTER_SCATTER_UPDATE_GPU(type) REGISTER_SCATTER_UPDATE(type, GPU);

Completed in 856 milliseconds