Home
last modified time | relevance | path

Searched refs:MaybeForwardRefInputToRefOutput (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops.cc994 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
1057 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
1288 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in DoCompute()
1403 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
1547 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
1619 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
1720 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
1836 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
1950 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
2083 MaybeForwardRefInputToRefOutput(ctx, 0, 0); in Compute()
[all …]
H A Dtraining_op_helpers.cc23 void MaybeForwardRefInputToRefOutput(OpKernelContext* ctx, int input, in MaybeForwardRefInputToRefOutput() function
H A Dtraining_op_helpers.h188 void MaybeForwardRefInputToRefOutput(OpKernelContext* ctx, int input,