Searched refs:tensor_for_ref (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | op_kernel.h | 464 void set_ref(int i, mutex* mu, Tensor* tensor_for_ref); 1010 Status set_output_ref(StringPiece name, mutex* mu, Tensor* tensor_for_ref); 1174 void set_output_ref(int index, mutex* mu, Tensor* tensor_for_ref); 1669 inline void OpOutputList::set_ref(int i, mutex* mu, Tensor* tensor_for_ref) { in set_ref() argument 1672 ctx_->set_output_ref(i, mu, tensor_for_ref); in set_ref()
|
H A D | op_kernel.cc | 1005 Tensor* tensor_for_ref) { in set_output_ref() argument 1009 outputs_[index] = TensorValue(mu, tensor_for_ref); in set_output_ref() 1013 Tensor* tensor_for_ref) { in set_output_ref() argument 1016 set_output_ref(index, mu, tensor_for_ref); in set_output_ref()
|