MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / set_host_tensor

Method set_host_tensor

tensorflow/compiler/jit/xla_tensor.h:87–89  ·  view source on GitHub ↗

Sets the contained host tensor.

Source from the content-addressed store, hash-verified

85 const Tensor& host_tensor() const { return *host_tensor_; }
86 // Sets the contained host tensor.
87 void set_host_tensor(const Tensor& tensor) {
88 host_tensor_.reset(new Tensor(tensor));
89 }
90
91 // Adds synchronization events to 'stream' that wait for this tensor to be
92 // defined on 'stream'. Does nothing if the tensor is already defined on that

Callers 2

CopyCPUTensorToDeviceMethod · 0.80
PopulateOutputsMethod · 0.80

Calls 1

resetMethod · 0.45

Tested by

no test coverage detected