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()
298 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
367 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
452 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
511 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
559 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &var)); in Compile()
581 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
|
| D | if_op.cc | 66 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(i + 1, &resource)); in Compile()
117 ctx->GetResourceInput(update.input_index + 1, &resource)); in Compile()
215 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
265 ctx->GetResourceInput(update.input_index + 1, &resource)); in Compile()
|
| D | case_op.cc | 63 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(i + 1, &resource)); in Compile()
116 ctx->GetResourceInput(update.input_index + 1, &resource)); in Compile()
227 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
277 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 | 60 TF_RETURN_IF_ERROR(ctx->GetResourceInput(i, &resource)); in MakeXlaCompilerArgumentsFromInputs()
351 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(update.input_index, &resource)); in Compile()
480 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(input_num, &resource)); in Compile()
556 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(update.input_index, &resource)); in Compile()
|
| D | variable_ops.cc | 35 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &variable)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 190 Status GetResourceInput(int index, XlaResource** resource);
|
| D | xla_op_kernel.cc | 500 Status XlaOpKernelContext::GetResourceInput(int index, XlaResource** resource) { in GetResourceInput() function in tensorflow::XlaOpKernelContext
|