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

Method really_record_tensor_reference

tensorflow/core/framework/op_kernel.cc:345–349  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

343}
344
345void OpKernelContext::really_record_tensor_reference(const Tensor& tensor) {
346 mutex_lock l(mu_);
347 // Keep a reference to the underlying memory around.
348 referenced_tensors_->Add(tensor);
349}
350
351Status OpKernelContext::input(StringPiece name, const Tensor** tensor) {
352 int start, stop;

Callers

nothing calls this directly

Calls 1

AddMethod · 0.45

Tested by

no test coverage detected