Home
last modified time | relevance | path

Searched refs:forward_ref_input_to_ref_output (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops.cc36 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()
Dtraining_op_helpers.cc26 ctx->forward_ref_input_to_ref_output(input, output); in MaybeForwardRefInputToRefOutput()
Didentity_op.h29 context->forward_ref_input_to_ref_output(0, 0); in Compute()
Dassign_op.h51 context->forward_ref_input_to_ref_output(0, 0); in Compute()
Ddense_update_ops.cc58 context->forward_ref_input_to_ref_output(0, 0); in Compute()
Dlogging_ops.cc92 ctx->forward_ref_input_to_ref_output(0, 0); in Compute()
Dscatter_op.cc115 c->forward_ref_input_to_ref_output(0, 0); in DoCompute()
Dshape_ops.cc489 ctx->forward_ref_input_to_ref_output(0, 0); in Compute()
Dstrided_slice_op.cc333 context->forward_ref_input_to_ref_output(0, 0); in Compute()
Dscatter_nd_op.cc289 c->forward_ref_input_to_ref_output(0, 0); in DoCompute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h870 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); in ForwardTfTensorInToOut()
886 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); in ForwardMklTensorInToOut()
887 context->forward_ref_input_to_ref_output(idx_meta_in, idx_meta_out); in ForwardMklTensorInToOut()
919 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out);
945 context->forward_ref_input_to_ref_output(idx_meta_in, idx_meta_out); in ForwardMklMetaDataInToOut()
/external/tensorflow/tensorflow/c/
Dkernels_experimental.cc331 cc_ctx->forward_ref_input_to_ref_output(input_index, output_index); in TF_OpKernelContext_ForwardRefInputToRefOutput()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h812 void forward_ref_input_to_ref_output(int input_index, int output_index);
Dop_kernel.cc433 void OpKernelContext::forward_ref_input_to_ref_output(int input_index, in forward_ref_input_to_ref_output() function in tensorflow::OpKernelContext