Home
last modified time | relevance | path

Searched refs:tensor_for_ref (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.h464 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 Dop_kernel.cc1005 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()