| 497 | break |
| 498 | |
| 499 | Status SetTensorValue(DataType dtype, int value, Tensor* tensor) { |
| 500 | // TODO(rmlarsen): Support more general shapes. |
| 501 | // TODO(lyandy): Change `value` to be int64 once int64 -> qint32 is supported. |
| 502 | if (tensor->NumElements() != 1) { |
| 503 | return errors::InvalidArgument( |
| 504 | "Expected scalar tensor, got num_elements = ", tensor->NumElements()); |
| 505 | } |
| 506 | switch (dtype) { |
| 507 | HANDLE_DOUBLE_CASE(DT_HALF); |
| 508 | HANDLE_DOUBLE_CASE(DT_BFLOAT16); |
| 509 | HANDLE_DOUBLE_CASE(DT_BOOL); |
| 510 | HANDLE_DOUBLE_CASE(DT_FLOAT); |
| 511 | HANDLE_DOUBLE_CASE(DT_DOUBLE); |
| 512 | HANDLE_DOUBLE_CASE(DT_UINT8); |
| 513 | HANDLE_DOUBLE_CASE(DT_INT8); |
| 514 | HANDLE_DOUBLE_CASE(DT_UINT16); |
| 515 | HANDLE_DOUBLE_CASE(DT_INT16); |
| 516 | HANDLE_DOUBLE_CASE(DT_INT32); |
| 517 | HANDLE_DOUBLE_CASE(DT_INT64); |
| 518 | HANDLE_DOUBLE_CASE(DT_COMPLEX64); |
| 519 | HANDLE_DOUBLE_CASE(DT_COMPLEX128); |
| 520 | HANDLE_INT_CASE(DT_QINT8); |
| 521 | HANDLE_INT_CASE(DT_QUINT8); |
| 522 | HANDLE_INT_CASE(DT_QINT16); |
| 523 | HANDLE_INT_CASE(DT_QUINT16); |
| 524 | HANDLE_INT_CASE(DT_QINT32); |
| 525 | default: |
| 526 | return errors::InvalidArgument("Unsupported type ", |
| 527 | DataTypeString(dtype)); |
| 528 | } |
| 529 | return Status::OK(); |
| 530 | } |
| 531 | |
| 532 | #undef HANDLE_CASE |
| 533 | |