Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc53 #define REGISTER_OPTIONAL_COPY(DIRECTION) \ macro
57 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
58 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
59 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc174 #define REGISTER_OPTIONAL_COPY(DIRECTION) \ macro
179 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
180 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
181 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);