Searched refs:from_allocation (Results 1 – 2 of 2) sorted by relevance
138 def from_allocation(cls, alloc: _ExtraFields_Allocation) -> Optional["TensorKey"]: member in TensorKey367 key = TensorKey.from_allocation(alloc_fields)470 key = TensorKey.from_allocation(i.typed[1])480 edges[TensorKey.from_allocation(i.typed[1])].input_version = None690 tkey = TensorKey.from_allocation(alloc_fields)
430 key = TensorKey.from_allocation(event.extra_fields)