Searched refs:replace_ref_input (Results 1 – 4 of 4) 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/contrib/framework/kernels/ |
| D | zero_initializer_op.cc | 56 ctx->replace_ref_input(0, *out_tensor, true); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 425 void OpKernelContext::replace_ref_input(int index, const Tensor& tensor, in replace_ref_input() function in tensorflow::OpKernelContext
617 Status OpKernelContext::replace_ref_input(StringPiece name, in replace_ref_input() function in tensorflow::OpKernelContext
631 replace_ref_input(start, tensor, lock_held); in replace_ref_input()
|
| D | op_kernel.h | 770 void replace_ref_input(int index, const Tensor& tensor, bool lock_held);
776 Status replace_ref_input(StringPiece name, const Tensor& tensor,
|