Searched refs:TensorReference (Results 1 – 15 of 15) sorted by relevance
30 class TensorReference {33 explicit TensorReference(const Tensor& tensor) in TensorReference() function38 ~TensorReference() {} in ~TensorReference()
675 friend class TensorReference; // For access to buf_. variable
842 TensorReference tensor_ref(*t); in allocate_persistent()
164 TensorReference tensor_ref(tensor); in SetProtoFromGPU()236 TensorReference input_ref(*input); in DeviceToDeviceCopy()286 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU()334 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU()
36 TensorReference reference;41 explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {} in TfDlManagedTensorCtx()270 TensorReference tensor_ref(*tensor); // This will call buf_->Ref() in TFE_HandleToDLPack()
189 TensorReference ref(*cpu_tensor); in CopyCPUTensorToDevice()252 TensorReference ref(*device_tensor); in CopyDeviceTensorToCPU()
280 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()
310 TensorReference partition_ref(partition_count); in ComputeAsync()
92 std::vector<TensorReference> scratch_tensor_refs_;
368 std::vector<TensorReference> scratch_tensor_refs_;
174 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()
850 TensorReference isvalid_dev_ref(isvalid_dev_tensor); in RunIfBoxIndexIsValid()