Searched refs:tensor_for_ref (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 500 void set_ref(int i, mutex* mu, Tensor* tensor_for_ref);
1007 Status set_output_ref(StringPiece name, mutex* mu, Tensor* tensor_for_ref);
1182 void set_output_ref(int index, mutex* mu, Tensor* tensor_for_ref);
1683 inline void OpOutputList::set_ref(int i, mutex* mu, Tensor* tensor_for_ref) { in set_ref() argument
1686 ctx_->set_output_ref(i, mu, tensor_for_ref); in set_ref()
|
| D | op_kernel.cc | 822 Tensor* tensor_for_ref) { in set_output_ref() argument
826 record_tensor_reference(*tensor_for_ref); in set_output_ref()
827 outputs_[index] = TensorValue(mu, tensor_for_ref); in set_output_ref()
831 Tensor* tensor_for_ref) { in set_output_ref() argument
840 set_output_ref(start, mu, tensor_for_ref); in set_output_ref()
|