Home
last modified time | relevance | path

Searched refs:forward_input_or_allocate_output (Results 1 – 25 of 41) 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.h902 Status forward_input_or_allocate_output(
906 Status forward_input_or_allocate_output(
1636 inline Status OpKernelContext::forward_input_or_allocate_output( in forward_input_or_allocate_output() function
1654 inline Status OpKernelContext::forward_input_or_allocate_output( in forward_input_or_allocate_output() function
/external/tensorflow/tensorflow/core/kernels/
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()
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.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()
301 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()
Dsoftmax_op.cc61 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()
Dcwise_ops_common.h100 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
113 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
127 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
274 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
305 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
Dconstant_op.cc249 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
296 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
Dxent_op.cc75 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dcollective_ops.cc284 c->forward_input_or_allocate_output( in ComputeAsyncImpl()
359 c->forward_input_or_allocate_output( in ComputeAsyncImpl()
549 c, c->forward_input_or_allocate_output({0}, 0, input.shape(), &output), in ComputeAsync()
812 c, c->forward_input_or_allocate_output({0}, 0, input.shape(), &output), in ComputeAsync()
941 c->forward_input_or_allocate_output( in ComputeAsync()
Disotonic_regression_op.cc189 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()
Dnccl_ops.cc101 c, c->forward_input_or_allocate_output({0}, 0, input->shape(), &output), in ComputeAsync()
Dbias_op.cc128 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
330 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()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dcholesky_op.cc127 context->forward_input_or_allocate_output( in ComputeAsync()
Dlu_op.cc101 context, context->forward_input_or_allocate_output( in Compute()
Dlu_op_gpu.cu.cc126 context->forward_input_or_allocate_output( in ComputeAsync()
Dmatrix_inverse_op.cc123 context->forward_input_or_allocate_output( in ComputeAsync()
Dmatrix_band_part_op.cc101 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
Dmatrix_set_diag_op.cc160 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
/external/tensorflow/tensorflow/core/kernels/image/
Dadjust_hue_op.cc70 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()

12