Home
last modified time | relevance | path

Searched refs:CopyTensorToDevice (Results 1 – 7 of 7) sorted by relevance

/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
300 input_buffer = CopyTensorToDevice(device, *input); in RunCollective()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcustom_device.h38 virtual Status CopyTensorToDevice(
H A Dcustom_device_op_handler.cc106 *status = dev->CopyTensorToDevice(handle, &result); in CopyTensorHandleToDevice()
H A Dcustom_device_test.cc38 Status CopyTensorToDevice(ImmediateExecutionTensorHandle* tensor, in CopyTensorToDevice() function in tensorflow::eager::__anonb9701f0e0111::TestCustomDevice
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor.h131 LazyTensorPtr CopyTensorToDevice(const BackendDevice& device);
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