| 80 | } |
| 81 | |
| 82 | Status LookupInterface::CheckFindArguments(const Tensor& key, |
| 83 | const Tensor& default_value) { |
| 84 | TF_RETURN_IF_ERROR(CheckKeyAndValueTypes(key, default_value)); |
| 85 | TF_RETURN_IF_ERROR(CheckKeyShape(key.shape())); |
| 86 | if (default_value.shape() != value_shape()) { |
| 87 | return errors::InvalidArgument( |
| 88 | "Expected shape ", value_shape().DebugString(), |
| 89 | " for default value, got ", default_value.shape().DebugString()); |
| 90 | } |
| 91 | return Status::OK(); |
| 92 | } |
| 93 | |
| 94 | } // namespace lookup |
| 95 | } // namespace tensorflow |
no test coverage detected