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

  /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
ring.cc 58 void CopyTensorData<CPUDevice>(void* dst, void* src, size_t size) {
ring.cu.cc 81 void CopyTensorData<GPUDevice>(void* dst, void* src, size_t size) {
ring.h 62 void CopyTensorData(void* destination, void* source, size_t size);
165 CopyTensorData<Device>((void*)buffer, (void*)input->tensor_data().data(),
282 CopyTensorData<Device>((void*)(buffer + offset),

Completed in 67 milliseconds