Home
last modified time | relevance | path

Searched defs:CopyTensorToDevice (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcustom_device_test.cc38 Status CopyTensorToDevice(ImmediateExecutionTensorHandle* tensor, in CopyTensorToDevice() function in tensorflow::eager::__anonb9701f0e0111::TestCustomDevice
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_test_util.cc262 Tensor CopyTensorToDevice(Device* device, const Tensor& tensor) { in CopyTensorToDevice() function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor.cpp311 LazyTensorPtr LazyTensor::CopyTensorToDevice(const BackendDevice& device) { in CopyTensorToDevice() function in torch::lazy::LazyTensor
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api.cc424 tensorflow::Status CopyTensorToDevice( in CopyTensorToDevice() function in tensorflow::__anon816881a00211::CustomDeviceAPI