Source code for afe.apis.compilation_job_base

#########################################################
# Copyright (C) 2022 SiMa Technologies, Inc.
#
# This material is SiMa proprietary and confidential.
#
# This material may not be copied or distributed without
# the express prior written permission of SiMa.
#
# All rights reserved.
#########################################################
# Code owner: Christopher Rodrigues
#########################################################
"""
Shared data type definitions related to compilation jobs.
"""
from typing import List, TypeVar

import numpy as np

# In Python code, tensor values are represented using Numpy arrays.


Tensor = np.ndarray


# Multiple tensors.  Used, for instance, to represent a model's inputs or outputs.


Tensors = List[Tensor]


# Ground truth data, representing the correct output of a model.


GroundTruth = TypeVar('GroundTruth')