Searched refs:replace_ref_input (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | ref_var.cc | 80 context->replace_ref_input(input_ref_index, reshaped_old_lhs, in AssignRefVariable()
94 context->replace_ref_input(input_ref_index, *input_alias, in AssignRefVariable()
108 context->replace_ref_input(input_ref_index, copy_tensor, in AssignRefVariable()
|
| H A D | op_kernel.cc | 455 void OpKernelContext::replace_ref_input(int index, const Tensor& tensor, in replace_ref_input() function in tensorflow::OpKernelContext
657 Status OpKernelContext::replace_ref_input(StringPiece name, in replace_ref_input() function in tensorflow::OpKernelContext
666 replace_ref_input(index, tensor, lock_held); in replace_ref_input()
|
| H A D | op_kernel.h | 782 void replace_ref_input(int index, const Tensor& tensor, bool lock_held);
788 Status replace_ref_input(StringPiece name, const Tensor& tensor,
|