Home
last modified time | relevance | path

Searched defs:vdot (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DBlas.cpp193 Tensor vdot(const Tensor &self, const Tensor &other){ in vdot() function
/external/pytorch/aten/src/ATen/cuda/
DCUDABlas.h279 inline void vdot(CUDABLAS_DOT_ARGTYPES(Dtype)) { in vdot() function
DCUDABlas.cpp1977 void vdot<c10::complex<float>>(CUDABLAS_DOT_ARGTYPES(c10::complex<float>)) { in vdot() function
1984 void vdot<c10::complex<double>>(CUDABLAS_DOT_ARGTYPES(c10::complex<double>)) { in vdot() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dasm_compiler.cc87 int64_t vmaj, vmin, vdot; in WarnIfBadPtxasVersion() local
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_math_ops.py335 def vdot(a, b): # pylint: disable=missing-docstring function
/external/pytorch/torch/_numpy/
D_funcs_impl.py1126 def vdot(a: ArrayLike, b: ArrayLike, /): function
/external/pytorch/torch/_refs/
D__init__.py6203 def vdot(self, other): function