config.external_configurable(torch.distributions.pareto.Pareto, module='torch.distributions.pareto') config.external_configurable(torch.distributions.poisson.Poisson, module='torch.distributions.poisson') config.external_configurable( torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli, module='torch.distributions.relaxed_bernoulli') config.external_configurable( torch.distributions.relaxed_bernoulli.RelaxedBernoulli, module='torch.distributions.relaxed_bernoulli') config.external_configurable( torch.distributions.relaxed_categorical.RelaxedOneHotCategorical, module='torch.distributions.relaxed_categorical') config.external_configurable(torch.distributions.studentT.StudentT, module='torch.distributions.studentT') config.external_configurable(torch.distributions.uniform.Uniform, module='torch.distributions.uniform') config.external_configurable(torch.distributions.weibull.Weibull, module='torch.distributions.weibull') # Constants config.constant('torch.float16', torch.float16) config.constant('torch.float32', torch.float32) config.constant('torch.float64', torch.float64) config.constant('torch.int8', torch.int8) config.constant('torch.uint8', torch.uint8) config.constant('torch.int16', torch.int16) config.constant('torch.int32', torch.int32) config.constant('torch.int64', torch.int64)
config.external_configurable(tf.random.stateless_categorical, 'tf.random.stateless_categorical') config.external_configurable(tf.random.stateless_normal, 'tf.random.stateless_normal') config.external_configurable(tf.random.stateless_truncated_normal, 'tf.random.stateless_truncated_normal') config.external_configurable(tf.random.stateless_uniform, 'tf.random.stateless_uniform') # Distribution strategies. config.external_configurable(tf.contrib.distribute.MirroredStrategy, module='tf.contrib.distribute') # Constants config.constant('tf.float16', tf.float16) config.constant('tf.float32', tf.float32) config.constant('tf.float64', tf.float64) config.constant('tf.bfloat16', tf.bfloat16) config.constant('tf.complex64', tf.complex64) config.constant('tf.complex128', tf.complex128) config.constant('tf.int8', tf.int8) config.constant('tf.uint8', tf.uint8) config.constant('tf.uint16', tf.uint16) config.constant('tf.int16', tf.int16) config.constant('tf.int32', tf.int32) config.constant('tf.int64', tf.int64) config.constant('tf.bool', tf.bool) config.constant('tf.string', tf.string) config.constant('tf.qint8', tf.qint8) config.constant('tf.quint8', tf.quint8)
config.external_configurable(nn.MultiheadAttention, 'multihead_att', module='T.nn') config.external_configurable(nn.PReLU, 'prelu', module='T.nn') config.external_configurable(nn.ReLU, 'relu', module='T.nn') config.external_configurable(nn.RReLU, 'rrelu', module='T.nn') config.external_configurable(nn.SELU, 'selu', module='T.nn') config.external_configurable(nn.CELU, 'celu', module='T.nn') config.external_configurable(nn.Sigmoid, 'sigmoid', module='T.nn') config.external_configurable(nn.Softplus, 'softplus', module='T.nn') config.external_configurable(nn.Softshrink, 'softshrink', module='T.nn') config.external_configurable(nn.Softsign, 'softsign', module='T.nn') config.external_configurable(nn.Tanh, 'tanh', module='T.nn') config.external_configurable(nn.Tanhshrink, 'tanhshrink', module='T.nn') config.external_configurable(nn.Threshold, 'threshold', module='T.nn') # constants config.constant('float16', T.float16) config.constant('float32', T.float32) config.constant('float64', T.float64) config.constant('int8', T.int8) config.constant('int16', T.int16) config.constant('int32', T.int32) config.constant('int64', T.int64) config.constant('complex32', T.complex32) config.constant('complex64', T.complex64) config.constant('complex128', T.complex128) config.constant('float', T.float) config.constant('short', T.short) config.constant('long', T.long) config.constant('half', T.half) config.constant('uint8', T.uint8) config.constant('int', T.int)