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()
|