Home
last modified time | relevance | path

Searched refs:DeviceMemoryFromTensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_tensor.h46 static se::DeviceMemoryBase DeviceMemoryFromTensor(const Tensor& tensor);
Dxla_tensor.cc30 /*static*/ se::DeviceMemoryBase XlaTensor::DeviceMemoryFromTensor( in DeviceMemoryFromTensor() function in tensorflow::XlaTensor
Dxla_launch_util.cc271 se::DeviceMemoryBase dmem = XlaTensor::DeviceMemoryFromTensor(*t); in PopulateInputs()