Searched refs:forward_ref_input_to_ref_output (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | control_flow_ops.cc | 36 context->forward_ref_input_to_ref_output(0, port); in Compute()
178 context->forward_ref_input_to_ref_output(index + 1, 0); in Compute()
220 context->forward_ref_input_to_ref_output(i, 0); in Compute()
297 context->forward_ref_input_to_ref_output(0, 0); in Compute()
354 context->forward_ref_input_to_ref_output(0, 0); in Compute()
406 context->forward_ref_input_to_ref_output(0, 0); in Compute()
|
| H A D | training_op_helpers.cc | 26 ctx->forward_ref_input_to_ref_output(input, output); in MaybeForwardRefInputToRefOutput()
|
| H A D | identity_op.h | 29 context->forward_ref_input_to_ref_output(0, 0); in Compute()
|
| H A D | dense_update_ops.cc | 58 context->forward_ref_input_to_ref_output(0, 0); in Compute()
|
| H A D | logging_ops.cc | 93 ctx->forward_ref_input_to_ref_output(0, 0); in Compute()
|
| H A D | scatter_op.cc | 125 c->forward_ref_input_to_ref_output(0, 0); in DoCompute()
|
| H A D | shape_ops.cc | 490 ctx->forward_ref_input_to_ref_output(0, 0); in Compute()
|
| H A D | strided_slice_op.cc | 337 context->forward_ref_input_to_ref_output(0, 0); in Compute()
|
| H A D | scatter_nd_op.cc | 290 c->forward_ref_input_to_ref_output(0, 0); in DoCompute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | ref_var.cc | 29 context->forward_ref_input_to_ref_output(input_ref_index, output_ref_index); in AssignRefVariable()
|
| H A D | op_kernel.h | 828 void forward_ref_input_to_ref_output(int input_index, int output_index);
|
| H A D | op_kernel.cc | 469 void OpKernelContext::forward_ref_input_to_ref_output(int input_index, in forward_ref_input_to_ref_output() function in tensorflow::OpKernelContext
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 904 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); in ForwardTfTensorInToOut()
920 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); in ForwardMklTensorInToOut()
921 context->forward_ref_input_to_ref_output(idx_meta_in, idx_meta_out); in ForwardMklTensorInToOut()
953 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out);
979 context->forward_ref_input_to_ref_output(idx_meta_in, idx_meta_out); in ForwardMklMetaDataInToOut()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels_experimental.cc | 384 cc_ctx->forward_ref_input_to_ref_output(input_index, output_index); in TF_OpKernelContext_ForwardRefInputToRefOutput()
|