(self)
| 53 | return tf_item.TF_IdentifyImportantOps(self.tf_item, sort_topologically) |
| 54 | |
| 55 | def GetOpProperties(self): |
| 56 | ret_from_swig = tf_item.TF_GetOpProperties(self.tf_item) |
| 57 | properties = {} |
| 58 | for key, values in ret_from_swig.items(): |
| 59 | prop = [] |
| 60 | for value in values: |
| 61 | prop.append( |
| 62 | op_performance_data_pb2.OpInfo.TensorProperties.FromString(value)) |
| 63 | properties[key] = prop |
| 64 | return properties |
| 65 | |
| 66 | def GetColocationGroups(self): |
| 67 | """Return a list of hard colocation constraints. |