| 53 | } // namespace |
| 54 | |
| 55 | bool IsValid(const ObjectDef& def) { |
| 56 | return def.data_type != DataType::UNKNOWN && |
| 57 | def.data_layout != DataLayout::UNKNOWN && |
| 58 | def.object_type != ObjectType::UNKNOWN; |
| 59 | } |
| 60 | |
| 61 | ObjectType GetType(const TensorObject& object) { |
| 62 | return absl::visit(ObjectTypeGetter{}, object); |