Searched refs:GetResourceInput (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 205 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
255 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
299 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
368 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
453 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
512 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
560 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &var)); in Compile()
582 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
|
| D | case_op.cc | 103 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(i + 1, &resource)); in Compile()
176 ctx->GetResourceInput(update.input_index + 1, &resource)); in Compile()
291 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
346 ctx->GetResourceInput(update.input_index + 1, &resource)); in Compile()
|
| D | if_op.cc | 65 ctx->GetResourceInput(update.input_index + 1, &resource)); in PopulateTensorArrayGradients()
207 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(i + 1, &resource)); in Compile()
311 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
364 ctx->GetResourceInput(update.input_index + 1, &resource)); in Compile()
|
| D | stack_ops.cc | 138 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
184 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
|
| D | while_op.cc | 85 TF_RETURN_IF_ERROR(ctx->GetResourceInput(i, &resource)); in MakeXlaCompilerArgumentsFromInputs()
384 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(update.input_index, &resource)); in Compile()
513 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
623 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(update.input_index, &resource)); in Compile()
|
| D | variable_ops.cc | 49 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &variable)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 255 Status GetResourceInput(int index, XlaResource** resource);
|
| D | xla_op_kernel.cc | 688 Status XlaOpKernelContext::GetResourceInput(int index, XlaResource** resource) { in GetResourceInput() function in tensorflow::XlaOpKernelContext
|