| 622 | } |
| 623 | |
| 624 | void SetOpAttrScalarDefault( |
| 625 | TFE_Context* ctx, TFE_Op* op, const tensorflow::AttrValue& default_value, |
| 626 | const char* attr_name, |
| 627 | tensorflow::gtl::FlatMap<string, tensorflow::int64>* attr_list_sizes, |
| 628 | TF_Status* status) { |
| 629 | SetOpAttrValueScalar(ctx, op, default_value, attr_name, status); |
| 630 | if (default_value.value_case() == tensorflow::AttrValue::kI) { |
| 631 | (*attr_list_sizes)[attr_name] = default_value.i(); |
| 632 | } |
| 633 | } |
| 634 | |
| 635 | // start_index is the index at which the Tuple/List attrs will start getting |
| 636 | // processed. |
no test coverage detected