Searched refs:TensorReference (Results 1 – 14 of 14) sorted by relevance
30 class TensorReference {33 explicit TensorReference(const Tensor& tensor) in TensorReference() function38 ~TensorReference() {} in ~TensorReference()
689 friend class TensorReference; // For access to buf_. variable
158 TensorReference input_ref(*input); in DeviceToDeviceCopy()208 TensorReference input_ref(*device_tensor); in CopyPluggableDeviceTensorToCPU()257 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToPluggableDevice()
164 TensorReference tensor_ref(tensor); in SetProtoFromGPU()236 TensorReference input_ref(*input); in DeviceToDeviceCopy()298 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU()344 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU()
38 TensorReference reference;43 explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {} in TfDlManagedTensorCtx()289 TensorReference tensor_ref(*tensor); // This will call buf_->Ref() in TFE_HandleToDLPack()
191 TensorReference ref(*cpu_tensor); in CopyCPUTensorToDevice()254 TensorReference ref(*device_tensor); in CopyDeviceTensorToCPU()
288 TensorReference input_reference(*input); in TpuDeviceToDeviceCopy()
82 TensorReference tensor_ref(out_of_line_values_on_host_); in Finalize()
258 TensorReference abnormal_detected_ref(abnormal_detected); in ComputeAsync()
299 TensorReference partition_ref(partition_count); in ComputeAsync()
190 TensorReference nnz_per_batch_device_ref(nnz_per_batch_device_t); in ComputeAsync()
221 TensorReference nnz_per_batch_device_ref(nnz_per_batch_device_t); in ComputeAsync()
555 std::vector<TensorReference> scratch_tensor_refs_;
877 TensorReference isvalid_dev_ref(isvalid_dev_tensor); in RunIfBoxIndexIsValid()