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

Method HasRemoteMirror

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

Source from the content-addressed store, hash-verified

352}
353
354bool TensorHandle::HasRemoteMirror(Device* d) {
355 tf_shared_lock l(remote_mirrors_mutex_);
356 auto mirror = remote_mirrors_.find(d);
357 if (mirror != remote_mirrors_.end()) {
358 return true;
359 }
360
361 auto unshaped_mirror = unshaped_remote_mirrors_.find(d);
362 if (unshaped_mirror != unshaped_remote_mirrors_.end()) {
363 return true;
364 }
365
366 return false;
367}
368
369Status TensorHandle::AddUnshapedRemoteMirror(
370 std::unique_ptr<UnshapedRemoteTensorHandleData> t, Device* d) {

Callers 1

EagerCopyToDeviceFunction · 0.80

Calls 2

findMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected