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