Searched refs:replace_ref_input (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | assign_op.h | 93 context->replace_ref_input(0, reshaped_old_lhs, in Compute()
107 context->replace_ref_input(0, *input_alias, /* lock_held */ true); in Compute()
121 context->replace_ref_input(0, *copyTensor, /* lock_held */ true); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 442 void OpKernelContext::replace_ref_input(int index, const Tensor& tensor, in replace_ref_input() function in tensorflow::OpKernelContext
613 Status OpKernelContext::replace_ref_input(StringPiece name, in replace_ref_input() function in tensorflow::OpKernelContext
622 replace_ref_input(index, tensor, lock_held); in replace_ref_input()
|
| D | op_kernel.h | 801 void replace_ref_input(int index, const Tensor& tensor, bool lock_held);
807 Status replace_ref_input(StringPiece name, const Tensor& tensor,
|