Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgather_op.cc135 #define REGISTER_GATHER_ALL_INDICES(dev, type) \ macro
139 #define REGISTER_GATHER_CPU(type) REGISTER_GATHER_ALL_INDICES(CPU, type)
154 #define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type)
167 #undef REGISTER_GATHER_ALL_INDICES
Dresource_variable_ops.cc713 #define REGISTER_GATHER_ALL_INDICES(dev, type) \ macro
717 #define REGISTER_GATHER_CPU(type) REGISTER_GATHER_ALL_INDICES(CPU, type)
725 #define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type)
750 #undef REGISTER_GATHER_ALL_INDICES