BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
blackcat
neural_networks
functions
common.h
Go to the documentation of this file.
1
/*
2
* Caffe_Cuda.h
3
*
4
* Created on: Nov 10, 2019
5
* Author: joseph
6
*/
7
8
#ifndef BLACKCAT_TENSORS_NEURALNETWORKS_FUNCTIONS_COMMON_CUDA_H_
9
#define BLACKCAT_TENSORS_NEURALNETWORKS_FUNCTIONS_COMMON_CUDA_H_
10
11
#ifdef __CUDACC__
12
#include <cuda_runtime_api.h>
13
#include <cuda.h>
14
#endif
15
namespace
bc
{
16
namespace
caffe {
17
18
inline
bool
is_a_ge_zero_and_a_lt_b
(
int
a,
int
b) {
19
return
static_cast<
unsigned
>
(a) < static_cast<unsigned>(b);
20
}
21
22
}
23
}
24
25
#endif
/* CAFFE_CUDA_H_ */
bc::caffe::is_a_ge_zero_and_a_lt_b
bool is_a_ge_zero_and_a_lt_b(int a, int b)
Definition:
common.h:18
bc
The Evaluator determines if an expression needs to be greedily optimized.
Definition:
algorithms.h:22
Generated by
1.8.13