Home
last modified time | relevance | path

Searched refs:REGISTER_SCATTER_MINMAX (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscatter_op.cc261 #define REGISTER_SCATTER_MINMAX(type, dev) \ macro
273 #define REGISTER_SCATTER_MINMAX_CPU(type) REGISTER_SCATTER_MINMAX(type, CPU);
286 #define REGISTER_SCATTER_MINMAX_GPU(type) REGISTER_SCATTER_MINMAX(type, GPU);
301 #define REGISTER_SCATTER_MINMAX_SYCL(type) REGISTER_SCATTER_MINMAX(type, SYCL);
317 #undef REGISTER_SCATTER_MINMAX
Dresource_variable_ops.cc844 #define REGISTER_SCATTER_MINMAX(type, dev) \ macro
853 #define REGISTER_SCATTER_MINMAX_CPU(type) REGISTER_SCATTER_MINMAX(type, CPU);
869 #define REGISTER_SCATTER_MINMAX_GPU(type) REGISTER_SCATTER_MINMAX(type, GPU);
904 #undef REGISTER_SCATTER_MINMAX