Home
last modified time | relevance | path

Searched refs:forward_input_or_allocate_output (Results 1 – 25 of 41) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_norm_op.cc118 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
121 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
124 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
130 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Dcwise_op_select.cc101 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in ComputeBroadcasting()
115 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in ComputeElementwise()
187 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
322 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in operator ()()
H A Dsparse_xent_op.cc94 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
97 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Dcollective_ops.cc300 c->forward_input_or_allocate_output( in ComputeAsyncImpl()
375 c->forward_input_or_allocate_output( in ComputeAsyncImpl()
736 c->forward_input_or_allocate_output( in ComputeAsync()
836 c->forward_input_or_allocate_output( in ComputeAsync()
1207 c->forward_input_or_allocate_output( in ComputeAsync()
1255 c->forward_input_or_allocate_output( in ComputeAsync()
H A Dcwise_ops_common.h110 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
123 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
137 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
289 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
320 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
H A Dconstant_op.cc261 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
310 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
H A Dsnapshot_op.cc36 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Dsoftmax_op.cc61 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Dcwise_ops_common.cc81 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in BinaryOpState()
H A Dcwise_op_leakyrelu_bf16.cc96 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
H A Dbias_op.cc129 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
271 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Dxent_op.cc88 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Disotonic_regression_op.cc190 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Dmaxpooling_op.cc338 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
562 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
1142 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
1211 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
H A Dcwise_op_clip.cc49 ctx, ctx->forward_input_or_allocate_output({0}, 0, in0.shape(), &out)); in Compute()
H A Dnccl_ops.cc101 c, c->forward_input_or_allocate_output({0}, 0, input->shape(), &output), in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dnumeric_op.h60 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
81 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/
H A Dgru_ops.cc125 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
307 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
313 ctx, ctx->forward_input_or_allocate_output( in Compute()
H A Dlstm_ops.cc370 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
385 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
684 ctx, ctx->forward_input_or_allocate_output( in Compute()
695 ctx, ctx->forward_input_or_allocate_output( in Compute()
700 ctx, ctx->forward_input_or_allocate_output( in Compute()
705 ctx, ctx->forward_input_or_allocate_output( in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dlu_op.cc101 context, context->forward_input_or_allocate_output( in Compute()
H A Dcholesky_op_gpu.cu.cc130 context->forward_input_or_allocate_output( in ComputeAsync()
H A Dlu_op_gpu.cu.cc126 context->forward_input_or_allocate_output( in ComputeAsync()
H A Dmatrix_inverse_op.cc123 context->forward_input_or_allocate_output( in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_layer_norm_op.cc141 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dadjust_hue_op.cc70 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()

12