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