nnlib
GPU-accelerated, C/C++ neural network library.
location_verifiers.h
Go to the documentation of this file.
1 
8 #ifndef NNLIB_LOCATION_VERIFIERS_H
9 #define NNLIB_LOCATION_VERIFIERS_H
10 
11 #include <tensor.h>
12 
19 bool allLocationsAreSame(std::initializer_list<DataLocation> locations);
20 
27 bool allLocationsAreHost(std::initializer_list<DataLocation> locations);
28 
35 bool allLocationsAreDevice(std::initializer_list<DataLocation> locations);
36 
37 #endif //NNLIB_LOCATION_VERIFIERS_H
bool allLocationsAreDevice(std::initializer_list< DataLocation > locations)
Return true if all given locations are DEVICE.
Definition: location_verifiers.cpp:29
bool allLocationsAreSame(std::initializer_list< DataLocation > locations)
Return true if all given locations are the same.
Definition: location_verifiers.cpp:11
bool allLocationsAreHost(std::initializer_list< DataLocation > locations)
Return true if all given locations are HOST.
Definition: location_verifiers.cpp:25
Header file declaring the Tensor class to represent multidimensional arrays.