DeviceTensor (definition) in projects: dports

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched defs:DeviceTensor (Results 1 – 5 of 5) sorted by relevance

/dports/math/faiss/faiss-1.7.1/faiss/gpu/utils/
H A D	DeviceTensor-inl.cuh	20 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor() in DeviceTensor() function in faiss::gpu::DeviceTensor 30 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor( in DeviceTensor() function in faiss::gpu::DeviceTensor 74 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor( in DeviceTensor() function in faiss::gpu::DeviceTensor 93 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor( in DeviceTensor() function in faiss::gpu::DeviceTensor 112 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor( in DeviceTensor() function in faiss::gpu::DeviceTensor 124 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor( in DeviceTensor() function in faiss::gpu::DeviceTensor 136 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor( in DeviceTensor() function in faiss::gpu::DeviceTensor 150 __host__ DeviceTensor<T, Dim, InnerContig, IndexT, PtrTraits>::DeviceTensor( in DeviceTensor() function in faiss::gpu::DeviceTensor
H A D	DeviceTensor.cuh	`22 class DeviceTensor : public Tensor<T, Dim, InnerContig, IndexT, PtrTraits> { class`
/dports/math/mfem/mfem-4.3/linalg/
H A D	dtensor.hpp	`81 class DeviceTensor class 94 DeviceTensor(Scalar* data_, Args... args) in DeviceTensor() function in mfem::DeviceTensor 104 MFEM_HOST_DEVICE DeviceTensor(const DeviceTensor& t) in DeviceTensor() function in mfem::DeviceTensor`
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A D	batch_norm.cu	425 struct DeviceTensor { struct 427 inline DeviceTensor() {} in DeviceTensor() argument 428 inline DeviceTensor(DType p, const int size) in DeviceTensor() function 435 MSHADOW_XINLINE unsigned getSize(const int i) const { in getSize() 439 MSHADOW_XINLINE int numElements() const { in numElements() 447 MSHADOW_XINLINE DType &operator()(const size_t batch, in operator ()() 464 MSHADOW_XINLINE DType &operator[](const size_t x) const { in operator []() 468 MSHADOW_XINLINE size_t InnerSize() const { in InnerSize() 476 MSHADOW_XINLINE size_t ChannelCount() const { in ChannelCount() 480 DType *dptr_; [all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A D	batch_norm.cu	425 struct DeviceTensor { struct 427 inline DeviceTensor() {} in DeviceTensor() argument 428 inline DeviceTensor(DType p, const int size) in DeviceTensor() argument 435 MSHADOW_XINLINE unsigned getSize(const int i) const { in getSize() 439 MSHADOW_XINLINE int numElements() const { in numElements() 447 MSHADOW_XINLINE DType &operator()(const size_t batch, in operator ()() 464 MSHADOW_XINLINE DType &operator[](const size_t x) const { in operator []() 468 MSHADOW_XINLINE size_t InnerSize() const { in InnerSize() 476 MSHADOW_XINLINE size_t ChannelCount() const { in ChannelCount() 480 DType *dptr_; [all …]