Searched defs:CopyTensorToDevice (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ | ||
| D | custom_device_test.cc | 38 Status CopyTensorToDevice(ImmediateExecutionTensorHandle* tensor, in CopyTensorToDevice() function in tensorflow::eager::__anonc9ab998e0111::TestCustomDevice |
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | collective_test_util.cc | 262 Tensor CopyTensorToDevice(Device* device, const Tensor& tensor) { in CopyTensorToDevice() function |
| /external/pytorch/torch/csrc/lazy/core/ | ||
| D | tensor.cpp | 311 LazyTensorPtr LazyTensor::CopyTensorToDevice(const BackendDevice& device) { in CopyTensorToDevice() function in torch::lazy::LazyTensor |
| /external/tensorflow/tensorflow/c/eager/ | ||
| D | c_api.cc | 424 tensorflow::Status CopyTensorToDevice( in CopyTensorToDevice() function in tensorflow::__anon7c894c200211::CustomDeviceAPI |