Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader.cc254 #define READER_COPY(dt) \ in GetTensor() macro
261 READER_COPY(DT_FLOAT); in GetTensor()
262 READER_COPY(DT_DOUBLE); in GetTensor()
263 READER_COPY(DT_INT32); in GetTensor()
264 READER_COPY(DT_UINT8); in GetTensor()
265 READER_COPY(DT_INT16); in GetTensor()
266 READER_COPY(DT_INT8); in GetTensor()
267 READER_COPY(DT_INT64); in GetTensor()
268 READER_COPY(DT_STRING); in GetTensor()
272 #undef READER_COPY in GetTensor()
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.cc218 #define READER_COPY(T) \ in RestoreTensor() macro
227 TF_CALL_SAVE_RESTORE_TYPES(READER_COPY) in RestoreTensor()
232 #undef READER_COPY in RestoreTensor()