Searched refs:set_host_tensor (Results 1 – 3 of 3) sorted by relevance
87 void set_host_tensor(const Tensor& tensor) { in set_host_tensor() function
136 xla_tensor->set_host_tensor(*cpu_tensor); in CopyCPUTensorToDevice()
342 xla_tensor->set_host_tensor(const_tensor); in PopulateOutputs()