Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.cc121 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()
127 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
133 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dsparse_xent_op.cc81 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
84 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dcwise_op_select.cc107 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in ComputeBroadcasting()
121 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in ComputeElementwise()
214 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in operator ()()
Dsnapshot_op.cc36 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dcwise_ops_common.cc69 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in BinaryOpState()
Dsoftmax_op.cc68 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dcholesky_op.cc128 context->forward_input_or_allocate_output( in ComputeAsync()
Dcollective_ops.cc201 c->forward_input_or_allocate_output( in ComputeAsync()
266 c->forward_input_or_allocate_output( in ComputeAsync()
Dconstant_op.cc273 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
330 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
Dmatrix_set_diag_op.cc84 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dxent_op.cc78 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dlu_op.cc101 context, context->forward_input_or_allocate_output( in Compute()
Dlu_op_gpu.cu.cc124 context->forward_input_or_allocate_output( in ComputeAsync()
Dmatrix_inverse_op.cc123 context->forward_input_or_allocate_output( in ComputeAsync()
Dsoftmax_op_gpu.cu.cc144 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dcwise_op_clip.cc49 ctx, ctx->forward_input_or_allocate_output({0}, 0, in0.shape(), &out)); in Compute()
Dmatrix_band_part_op.cc100 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dadjust_hue_op.cc70 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dmaxpooling_op.cc305 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
328 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in MaxPoolingBackwardCustomKernel()
530 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
1047 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
1101 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dbias_op.cc129 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
363 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dmatrix_solve_op.cc160 context->forward_input_or_allocate_output({1}, 0, rhs.shape(), &output), in ComputeAsync()
/external/tensorflow/tensorflow/core/framework/
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()
Dop_kernel.h858 Status forward_input_or_allocate_output(
861 Status forward_input_or_allocate_output(
1585 inline Status OpKernelContext::forward_input_or_allocate_output( in forward_input_or_allocate_output() function
1597 inline Status OpKernelContext::forward_input_or_allocate_output( in forward_input_or_allocate_output() function
/external/tensorflow/tensorflow/contrib/rnn/kernels/
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()
Dlstm_ops.cc307 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
322 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
585 ctx, ctx->forward_input_or_allocate_output( in Compute()
596 ctx, ctx->forward_input_or_allocate_output( in Compute()
601 ctx, ctx->forward_input_or_allocate_output( in Compute()
606 ctx, ctx->forward_input_or_allocate_output( in Compute()

12