| 837 | } |
| 838 | |
| 839 | void TF_SetAttrInt(TF_OperationDescription* desc, const char* attr_name, |
| 840 | int64_t value) { |
| 841 | static_assert(sizeof(int64_t) == sizeof(tensorflow::int64), |
| 842 | "64-bit int types should match in size"); |
| 843 | desc->node_builder.Attr(attr_name, static_cast<tensorflow::int64>(value)); |
| 844 | } |
| 845 | |
| 846 | void TF_SetAttrIntList(TF_OperationDescription* desc, const char* attr_name, |
| 847 | const int64_t* values, int num_values) { |