/external/tensorflow/tensorflow/core/kernels/ |
D | batch_norm_op.cc | 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() 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()
|
D | sparse_xent_op.cc | 81 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()
|
D | cwise_op_select.cc | 107 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 ()()
|
D | snapshot_op.cc | 36 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
|
D | cwise_ops_common.cc | 69 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in BinaryOpState()
|
D | softmax_op.cc | 68 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
|
D | cholesky_op.cc | 128 context->forward_input_or_allocate_output( in ComputeAsync()
|
D | collective_ops.cc | 201 c->forward_input_or_allocate_output( in ComputeAsync() 266 c->forward_input_or_allocate_output( in ComputeAsync()
|
D | constant_op.cc | 273 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()
|
D | matrix_set_diag_op.cc | 84 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
|
D | xent_op.cc | 78 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
|
D | lu_op.cc | 101 context, context->forward_input_or_allocate_output( in Compute()
|
D | lu_op_gpu.cu.cc | 124 context->forward_input_or_allocate_output( in ComputeAsync()
|
D | matrix_inverse_op.cc | 123 context->forward_input_or_allocate_output( in ComputeAsync()
|
D | softmax_op_gpu.cu.cc | 144 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
|
D | cwise_op_clip.cc | 49 ctx, ctx->forward_input_or_allocate_output({0}, 0, in0.shape(), &out)); in Compute()
|
D | matrix_band_part_op.cc | 100 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
|
D | adjust_hue_op.cc | 70 OP_REQUIRES_OK(context, context->forward_input_or_allocate_output( in Compute()
|
D | maxpooling_op.cc | 305 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()
|
D | bias_op.cc | 129 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()
|
D | matrix_solve_op.cc | 160 context->forward_input_or_allocate_output({1}, 0, rhs.shape(), &output), in ComputeAsync()
|
/external/tensorflow/tensorflow/core/framework/ |
D | numeric_op.h | 60 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()
|
D | op_kernel.h | 858 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/ |
D | gru_ops.cc | 125 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()
|
D | lstm_ops.cc | 307 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()
|