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

Method TensorHandle

tensorflow/core/common_runtime/eager/tensor_handle.cc:111–128  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

109}
110
111TensorHandle::TensorHandle(std::unique_ptr<LocalTensorHandleData> t,
112 DataType dtype, Device* d, Device* op_device,
113 EagerContext* ctx)
114 : dtype(dtype),
115 device_(d),
116 op_device_(op_device),
117 resource_device_(nullptr),
118#if !defined(IS_MOBILE_PLATFORM)
119 remote_op_id_(kInvalidOpId),
120 remote_output_num_(kInvalidOutputNum),
121#endif
122 ctx_(ctx),
123 is_remote_(false),
124 tensor_handle_data_(std::move(t)) {
125 VLOG(3) << "Creating Local TensorHandle: " << this << " device: " << device_;
126 // Notify immediately since this handle is already ready.
127 is_ready_notification_.Notify();
128}
129
130TensorHandle::TensorHandle(std::unique_ptr<LocalTensorHandleData> t,
131 const ResourceHandle& resource_handle, Device* d,

Callers 1

build_resultsMethod · 0.45

Calls 7

GetResourceDeviceFunction · 0.85
dtypes_and_shapesMethod · 0.80
eager_clientMethod · 0.80
context_idMethod · 0.80
NotifyMethod · 0.45
op_idMethod · 0.45
output_numMethod · 0.45

Tested by

no test coverage detected