Searched refs:DeviceMemoryFromTensor (Results 1 – 3 of 3) sorted by relevance
46 static se::DeviceMemoryBase DeviceMemoryFromTensor(const Tensor& tensor);
30 /*static*/ se::DeviceMemoryBase XlaTensor::DeviceMemoryFromTensor( in DeviceMemoryFromTensor() function in tensorflow::XlaTensor
271 se::DeviceMemoryBase dmem = XlaTensor::DeviceMemoryFromTensor(*t); in PopulateInputs()