mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00
remove stray file
This commit is contained in:
parent
5e9fcbb6e8
commit
377322d3d4
@ -1,52 +0,0 @@
|
||||
# Copyright 2018 Google LLC
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# https://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
import numpy as onp
|
||||
import scipy.stats as osp_stats
|
||||
|
||||
from .. import lax
|
||||
|
||||
|
||||
def _wraps(fun):
|
||||
"""Like functools.wraps but works with numpy.ufuncs."""
|
||||
docstr = """
|
||||
LAX-backed implementation of {fun}. Corresponding Scipy docstring below.
|
||||
|
||||
{np_doc}
|
||||
""".format(fun=fun.__name__, np_doc=fun.__doc__)
|
||||
def wrap(op):
|
||||
try:
|
||||
op.__name__ = fun.__name__
|
||||
op.__doc__ = docstr
|
||||
finally:
|
||||
return op
|
||||
return wrap
|
||||
|
||||
|
||||
@_wraps(osp_stats.norm.logpdf)
|
||||
def norm_logpdf(x, loc=0, scale=1):
|
||||
x, loc, scale = _promote_args_like(osp_stats.norm.logpdf, x, loc, scale)
|
||||
two = _constant_like(x, 2)
|
||||
scale_sqrd = lax.pow(scale, two)
|
||||
log_normalizer = lax.log(lax.mul(_constant_like(x, 2 * onp.pi), scale_sqrd))
|
||||
quadratic = lax.div(lax.pow(lax.sub(x, loc), two), scale_sqrd)
|
||||
return lax.div(lax.neg(lax.add(log_normalizer, quadratic)), two)
|
||||
|
||||
|
||||
def _constant_like(x, const):
|
||||
return onp.array(const, dtype=lax._dtype(x))
|
Loading…
x
Reference in New Issue
Block a user