Searched defs:ufunc (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/test/torch_np/ |
| D | test_ufuncs_basic.py | 42 def get_x(self, ufunc): argument
46 def test_scalar(self, ufunc): argument
53 def test_x_and_dtype(self, ufunc): argument
62 def test_x_and_dtype_casting(self, ufunc, casting, dtype): argument
73 def test_x_and_out_casting(self, ufunc, casting, out_dtype): argument
85 def test_x_and_out_broadcast(self, ufunc): argument
164 def get_xy(self, ufunc): argument
168 def test_scalar(self, ufunc): argument
175 def test_vector_vs_scalar(self, ufunc): argument
182 def test_xy_and_out_casting(self, ufunc, casting, out_dtype): argument
[all …]
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseUnaryOps.cpp | 82 Tensor coalesced_unary_ufunc(const Tensor &self, const Ufunc &ufunc) { in coalesced_unary_ufunc()
98 Tensor& coalesced_unary_ufunc_(Tensor &self, const Ufunc &ufunc) { in coalesced_unary_ufunc_()
106 Tensor& coalesced_unary_ufunc_out(const Tensor &self, Tensor &result, const Ufunc &ufunc) { in coalesced_unary_ufunc_out()
|
| /external/ml_dtypes/ml_dtypes/tests/ |
| D | intn_test.py | 291 def testUnaryPredicateUfunc(self, scalar_type, ufunc): argument
314 def testPredicateUfuncs(self, scalar_type, ufunc): argument
379 def testBinaryUfuncs(self, scalar_type, ufunc): argument
|
| /external/pytorch/torch/_numpy/ |
| D | _ufuncs.py | 234 ufunc = getattr(_binary_ufuncs_impl, name) variable
330 ufunc = getattr(_unary_ufuncs_impl, name) variable
|
| /external/pytorch/torchgen/dest/ |
| D | ufunc.py | 6 import torchgen.api.ufunc as ufunc namespace
|
| /external/ml_dtypes/ml_dtypes/_src/ |
| D | ufuncs.h | 113 PyUFuncObject* ufunc = reinterpret_cast<PyUFuncObject*>(ufunc_obj.get()); in RegisterUFunc() local
|
| /external/pytorch/torchgen/ |
| D | model.py | 866 import torchgen.api.ufunc as ufunc namespace
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 1014 PyUFuncObject* ufunc = reinterpret_cast<PyUFuncObject*>(ufunc_obj.get()); in RegisterUFunc() local
|