Searched refs:input_ref_index (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | ref_var.cc | 22 OpKernelContext* context, int input_ref_index, int output_ref_index, in AssignRefVariable() argument
29 context->forward_ref_input_to_ref_output(input_ref_index, output_ref_index); in AssignRefVariable()
48 mutex_lock l(*context->input_ref_mutex(input_ref_index)); in AssignRefVariable()
50 context->mutable_input(input_ref_index, /* lock_held */ true); in AssignRefVariable()
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()
121 context->mutable_input(input_ref_index, /* lock_held */ false); in AssignRefVariable()
|
| H A D | ref_var.h | 25 OpKernelContext* context, int input_ref_index, int output_ref_index,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | assign_op.h | 49 constexpr int input_ref_index = 0; in Compute() local
56 AssignRefVariable(context, input_ref_index, output_ref_index, value_index, in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels_experimental.h | 77 TF_OpKernelContext* ctx, int input_ref_index, int output_ref_index,
|
| H A D | kernels_experimental.cc | 219 void TF_AssignRefVariable(TF_OpKernelContext* ctx, int input_ref_index, in TF_AssignRefVariable() argument
244 ::tensorflow::AssignRefVariable(cc_ctx, input_ref_index, output_ref_index, in TF_AssignRefVariable()
|