| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_rma_distributed.cc | 72 Tensor* cpu_tensor) { in PopulateTensorFromExtra()
82 Tensor* cpu_tensor) { in PopulateTensorFromResponse()
127 std::unique_ptr<Tensor> cpu_tensor; in RecvFromPeer() member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device_test.cc | 107 void InitCPUTensor(Tensor* cpu_tensor, int num_elements, float value) { in InitCPUTensor()
114 void CopyCPUToGPU(Tensor* cpu_tensor, Tensor* gpu_tensor, Device* device, in CopyCPUToGPU()
120 void CopyGPUToCPU(Tensor* gpu_tensor, Tensor* cpu_tensor, Device* device, in CopyGPUToCPU()
533 Tensor cpu_tensor(cpu_allocator(), DT_FLOAT, TensorShape({kNumElements})); in TEST_F() local
|
| H A D | gpu_util_platform_specific.cc | 26 void GPUDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
37 Device* device, Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| H A D | gpu_util.cc | 268 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU()
311 void GPUUtil::CopyCPUTensorToGPU(const Tensor* cpu_tensor, in CopyCPUTensorToGPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_context.cc | 28 const Tensor* cpu_tensor, Device* device, Tensor* device_tensor, in CopyCPUTensorToDevice()
37 Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| H A D | pluggable_device_util.cc | 176 const Tensor* device_tensor, Tensor* cpu_tensor, StatusCallback done) { in CopyPluggableDeviceTensorToCPU()
222 const Tensor* cpu_tensor, const DeviceContext* device_context, in CopyCPUTensorToPluggableDevice()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | Utils.cpp | 34 auto cpu_tensor = tensor_cpu(values, options.device(DeviceType::CPU)); in tensor_backend() local
53 auto cpu_tensor = tensor_complex_cpu(values, options.device(DeviceType::CPU)); in tensor_complex_backend() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | virtual_device.cc | 38 void VirtualDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
50 Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.cc | 40 Tensor* cpu_tensor) { in CopyDeviceTensorToCPUSync()
52 Status DeviceContext::CopyCPUTensorToDeviceSync(const Tensor* cpu_tensor, in CopyCPUTensorToDeviceSync()
|
| H A D | device_base.h | 101 Tensor* cpu_tensor, StatusCallback done) { in CopyDeviceTensorToCPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device_context.cc | 114 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
211 Device* device, Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | serialization.cpp | 231 at::Tensor cpu_tensor; in THPStorage_writeFileRaw() local
386 auto cpu_tensor = at::from_blob( in THPStorage_readFileRaw() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | stack.cc | 257 Tensor* cpu_tensor = in ComputeAsync() local
304 Tensor* cpu_tensor = &value.tensor; in ComputeAsync() local
|
| H A D | dynamic_partition_op_gpu.cu.cc | 276 Tensor cpu_tensor; in ComputeAsync() local
|
| H A D | collective_nccl_test.cc | 273 Tensor cpu_tensor(dtype, shape); in InitTensor() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | MKLDNNConversions.cpp | 51 Tensor cpu_tensor = at::empty( in mkldnn_to_dense() local
78 Tensor dense_to_mkldnn(const Tensor& cpu_tensor, std::optional<ScalarType> dtype) { in dense_to_mkldnn()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| H A D | op_handler_selector_test.cc | 401 tensorflow::core::RefCountPtr<FakeTensorHandle> cpu_tensor( in TEST_F() local
430 tensorflow::core::RefCountPtr<FakeTensorHandle> cpu_tensor( in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | ring_alg.cc | 272 Tensor cpu_tensor(tensor.dtype(), tensor.shape()); in TensorDebugString() local
|
| H A D | process_function_library_runtime_test.cc | 183 Tensor cpu_tensor(device_tensor.dtype(), device_tensor.shape()); in GPUToCPU() local
194 Tensor CPUToGPU(const Tensor& cpu_tensor) { in CPUToGPU()
|
| H A D | copy_tensor.cc | 255 Tensor* cpu_tensor = in ViaDMA() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| H A D | performance_profiling.cc | 122 TensorFloat32 cpu_tensor; in RunExternalImmutableSample() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_worker_service.cc | 681 Tensor* cpu_tensor = in RecvBufAsync() local
|