Searched defs:DefaultDevice (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorDeviceDefault.h | 17 struct DefaultDevice { struct
18 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void* allocate(size_t num_bytes) const { in allocate()
21 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void deallocate(void* buffer) const { in deallocate()
24 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void* allocate_temp(size_t num_bytes) const { in allocate_temp()
27 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void deallocate_temp(void* buffer) const { in deallocate_temp()
30 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void memcpy(void* dst, const void* src, size_t n) const { in memcpy()
33 …VICE_FUNC EIGEN_STRONG_INLINE void memcpyHostToDevice(void* dst, const void* src, size_t n) const { in memcpyHostToDevice()
36 …VICE_FUNC EIGEN_STRONG_INLINE void memcpyDeviceToHost(void* dst, const void* src, size_t n) const { in memcpyDeviceToHost()
39 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void memset(void* buffer, int c, size_t n) const { in memset()
43 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Type get(Type data) const { in get()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | virtual_placer_test.cc | 168 TEST(VirtualPlacerTest, DefaultDevice) { in TEST() argument
|
| /aosp_15_r20/external/deqp/external/vulkancts/modules/vulkan/ |
| H A D | vktTestCase.cpp | 446 class DefaultDevice class
693 DefaultDevice::DefaultDevice(const PlatformInterface &vkPlatform, const tcu::CommandLine &cmdLine, in DefaultDevice() function in vkt::DefaultDevice
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 80 static inline at::DeviceType DefaultDevice() { in DefaultDevice() function
|