Searched refs:TensorSignature (Results 1 – 1 of 1) sorted by relevance
37 class TensorSignature(collections.namedtuple( class65 return super(TensorSignature, cls).__new__(67 return super(TensorSignature, cls).__new__(133 if not TensorSignature(tensors[key]).is_compatible_with(signatures[key]):140 return TensorSignature(tensors).is_compatible_with(signatures)154 key: TensorSignature(tensors[key]) for key in tensors}157 return TensorSignature(tensors)