Home
last modified time | relevance | path

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

12

/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.h911 Status forward_input_or_allocate_output(
914 Status forward_input_or_allocate_output(
1685 inline Status OpKernelContext::forward_input_or_allocate_output( in forward_input_or_allocate_output() function
1697 inline Status OpKernelContext::forward_input_or_allocate_output( in forward_input_or_allocate_output() function
/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()
211 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
346 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.cc81 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()
Dcwise_ops_common.h106 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
119 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
133 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
280 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
311 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
Dconstant_op.cc290 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
348 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
Dxent_op.cc78 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dcollective_ops.cc233 c->forward_input_or_allocate_output( in ComputeAsync()
305 c->forward_input_or_allocate_output( in ComputeAsync()
Dlu_op.cc101 context, context->forward_input_or_allocate_output( in Compute()
Dcholesky_op.cc128 context->forward_input_or_allocate_output( in ComputeAsync()
Dcwise_op_clip.cc49 ctx, ctx->forward_input_or_allocate_output({0}, 0, in0.shape(), &out)); in Compute()
Dmatrix_inverse_op.cc123 context->forward_input_or_allocate_output( in ComputeAsync()
Dlu_op_gpu.cu.cc126 context->forward_input_or_allocate_output( in ComputeAsync()
Dmatrix_band_part_op.cc100 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dsoftmax_op_gpu.cu.cc148 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()
Dnccl_ops.cc101 c, c->forward_input_or_allocate_output({0}, 0, input->shape(), &output), in ComputeAsync()
Dbias_op.cc131 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
358 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dmaxpooling_op.cc307 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
330 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in MaxPoolingBackwardCustomKernel()
532 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
1049 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
1103 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
/external/tensorflow/tensorflow/core/kernels/rnn/
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.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()
625 ctx, ctx->forward_input_or_allocate_output( in Compute()
636 ctx, ctx->forward_input_or_allocate_output( in Compute()
641 ctx, ctx->forward_input_or_allocate_output( in Compute()
646 ctx, ctx->forward_input_or_allocate_output( in Compute()

12