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

  /external/tensorflow/tensorflow/core/kernels/
gather_op.cc 135 #define REGISTER_GATHER_ALL_INDICES(dev, type) \
139 #define REGISTER_GATHER_CPU(type) REGISTER_GATHER_ALL_INDICES(CPU, type)
152 #define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type)
162 #undef REGISTER_GATHER_ALL_INDICES
resource_variable_ops.cc 543 #define REGISTER_GATHER_ALL_INDICES(dev, type) \
547 #define REGISTER_GATHER_CPU(type) REGISTER_GATHER_ALL_INDICES(CPU, type)
555 #define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type)
563 #undef REGISTER_GATHER_ALL_INDICES

Completed in 80 milliseconds