| 381 | } |
| 382 | |
| 383 | Status OpKernelContext::input_ref_mutex(StringPiece name, mutex** out_mutex) { |
| 384 | int start, stop; |
| 385 | TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); |
| 386 | if (stop != start + 1) { |
| 387 | return errors::InvalidArgument("OpKernel used list-valued input name '", |
| 388 | name, |
| 389 | "' when single-valued input was expected"); |
| 390 | } |
| 391 | *out_mutex = input_ref_mutex(start); |
| 392 | return Status::OK(); |
| 393 | } |
| 394 | |
| 395 | const Tensor& OpKernelContext::input(int index) { |
| 396 | CHECK_GE(index, 0); |
nothing calls this directly
no test coverage detected