| 175 | } |
| 176 | |
| 177 | Status GetInputList(OpKernelContext* c, int index, const TensorList** list) { |
| 178 | if (!TensorShapeUtils::IsScalar(c->input(index).shape())) { |
| 179 | return errors::InvalidArgument("Input list must be a scalar saw: ", |
| 180 | c->input(index).shape().DebugString()); |
| 181 | } |
| 182 | const TensorList* l = c->input(index).scalar<Variant>()().get<TensorList>(); |
| 183 | if (l == nullptr) { |
| 184 | return errors::InvalidArgument( |
| 185 | "Input handle is not a list. Saw: '", |
| 186 | c->input(index).scalar<Variant>()().DebugString(), "'"); |
| 187 | } |
| 188 | *list = l; |
| 189 | return Status::OK(); |
| 190 | } |
| 191 | |
| 192 | Status ForwardInputOrCreateNewList(OpKernelContext* c, int32 input_index, |
| 193 | int32 output_index, |
no test coverage detected