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

  /external/tensorflow/tensorflow/core/kernels/
gather_op.cc 152 #define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type)
154 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GATHER_GPU);
155 TF_CALL_complex64(REGISTER_GATHER_GPU);
156 TF_CALL_complex128(REGISTER_GATHER_GPU);
158 #undef REGISTER_GATHER_GPU
resource_variable_ops.cc 555 #define REGISTER_GATHER_GPU(type) REGISTER_GATHER_ALL_INDICES(GPU, type)
557 TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_GATHER_GPU);
562 #undef REGISTER_GATHER_GPU

Completed in 650 milliseconds