Searched refs:record_tensor_reference (Results 1 – 2 of 2) sorted by relevance
365 record_tensor_reference(**tensor); in input()404 record_tensor_reference(tensor); in input()415 record_tensor_reference(tensor); in mutable_input()420 record_tensor_reference(tensor); in mutable_input()437 record_tensor_reference(tensor); in replace_ref_input()613 record_tensor_reference(*tensor); in mutable_input()720 record_tensor_reference(new_tensor); in allocate_tensor()801 record_tensor_reference(tensor); in set_output()826 record_tensor_reference(*tensor_for_ref); in set_output_ref()
1235 void record_tensor_reference(const Tensor& tensor);1533 inline void OpKernelContext::record_tensor_reference(const Tensor& tensor) { in record_tensor_reference() function1565 record_tensor_reference(t); in NotifyUseOfPersistentTensor()