Searched refs:record_tensor_reference (Results 1 – 2 of 2) sorted by relevance
327 record_tensor_reference(**tensor); in input()366 record_tensor_reference(tensor); in input()377 record_tensor_reference(tensor); in mutable_input()382 record_tensor_reference(tensor); in mutable_input()399 record_tensor_reference(tensor); in replace_ref_input()545 record_tensor_reference(*tensor); in mutable_input()644 record_tensor_reference(new_tensor); in allocate_tensor()725 record_tensor_reference(tensor); in set_output()750 record_tensor_reference(*tensor_for_ref); in set_output_ref()
1080 void record_tensor_reference(const Tensor& tensor);1337 inline void OpKernelContext::record_tensor_reference(const Tensor& tensor) { in record_tensor_reference() function1369 record_tensor_reference(t); in NotifyUseOfPersistentTensor()