Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgather_op.cc139 #define REGISTER_GATHER_CPU(type) REGISTER_GATHER_ALL_INDICES(CPU, type) macro
142 TF_CALL_ALL_TYPES(REGISTER_GATHER_CPU);
143 TF_CALL_QUANTIZED_TYPES(REGISTER_GATHER_CPU);
144 TF_CALL_quint16(REGISTER_GATHER_CPU);
145 TF_CALL_qint16(REGISTER_GATHER_CPU);
146 TF_CALL_uint32(REGISTER_GATHER_CPU);
147 TF_CALL_uint64(REGISTER_GATHER_CPU);
149 #undef REGISTER_GATHER_CPU
Dresource_variable_ops.cc717 #define REGISTER_GATHER_CPU(type) REGISTER_GATHER_ALL_INDICES(CPU, type) macro
720 TF_CALL_ALL_TYPES(REGISTER_GATHER_CPU);
721 TF_CALL_QUANTIZED_TYPES(REGISTER_GATHER_CPU);
748 #undef REGISTER_GATHER_CPU