def get_dtype(self):
"""Get data type info."""
dtype_idx = self.tensor_type
dtype_str = Input.DATA_TYPE_MAP[dtype_idx]
if DLK_DTYPE_MAP[dtype_str] is None:
raise UnsupportedDataType(f'Type {dtype_str} is not supported.')
return DLK_DTYPE_MAP[dtype_str]
def get_dtype(self) -> DataType:
"""Get data type info."""
dtype_idx = self.tensor_type.elem_type
dtype_str = TensorProtoWrapper.DATA_TYPE_MAP[dtype_idx]
if DLK_DTYPE_MAP[dtype_str] is None:
raise UnsupportedDataType(f'Type {dtype_str} is not supported.')
return DLK_DTYPE_MAP[dtype_str] # type: ignore
def get_dtype(self):
"""Get data type info."""
if self.is_placeholder:
dtype_str = type(self).DATA_TYPE_MAP[self.in_.attr.get(
'dtype').type]
else:
dtype_str = type(self).DATA_TYPE_MAP[self.tensor.dtype]
if DLK_DTYPE_MAP[dtype_str] is None:
raise UnsupportedDataType(f'Type {dtype_str} is not supported.')
return DLK_DTYPE_MAP[dtype_str]
def get_dtype(self):
"""Get dlk dtype of the node."""
dtype_str = Input.DATA_TYPE_MAP[self.tensor_type]
if DLK_DTYPE_MAP[dtype_str] is None:
raise UnsupportedDataType(f'Type {dtype_str} is not supported.')
return DLK_DTYPE_MAP[dtype_str]
