| 960 | } |
| 961 | |
| 962 | Status OpKernelContext::mutable_output(StringPiece name, Tensor** tensor) { |
| 963 | int start, stop; |
| 964 | TF_RETURN_IF_ERROR(params_->op_kernel->OutputRange(name, &start, &stop)); |
| 965 | if (stop != start + 1) { |
| 966 | return errors::InvalidArgument("OpKernel used list-valued output name '", |
| 967 | name, |
| 968 | "' when single-valued output was " |
| 969 | "expected"); |
| 970 | } |
| 971 | *tensor = mutable_output(start); |
| 972 | return Status::OK(); |
| 973 | } |
| 974 | |
| 975 | bool OpKernelContext::ValidateInputsAreSameShape(OpKernel* op) { |
| 976 | const auto& inputs = *params_->inputs; |
nothing calls this directly
no test coverage detected