Welcome to mirror list, hosted at ThFree Co, Russian Federation.

THCTensorMathCompareT.cu « generic « THC « lib - github.com/torch/cutorch.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 4b59abf1bf383ba9cf043cfa50c3c30d293e5c25 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCTensorMathCompareT.cu"
#else

THC_API void
THCTensor_(ltTensor)(THCState *state, THCudaByteTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorLTOp<typename TensorUtils<THCTensor>::DataType,
                    unsigned char>());
}

THC_API void
THCTensor_(gtTensor)(THCState *state, THCudaByteTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorGTOp<typename TensorUtils<THCTensor>::DataType,
                    unsigned char>());
}

THC_API void
THCTensor_(leTensor)(THCState *state, THCudaByteTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorLEOp<typename TensorUtils<THCTensor>::DataType,
                    unsigned char>());
}

THC_API void
THCTensor_(geTensor)(THCState *state, THCudaByteTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorGEOp<typename TensorUtils<THCTensor>::DataType,
                    unsigned char>());
}

THC_API void
THCTensor_(eqTensor)(THCState *state, THCudaByteTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorEQOp<typename TensorUtils<THCTensor>::DataType,
                    unsigned char>());
}

THC_API void
THCTensor_(neTensor)(THCState *state, THCudaByteTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorNEOp<typename TensorUtils<THCTensor>::DataType,
                    unsigned char>());
}

THC_API void
THCTensor_(ltTensorT)(THCState *state, THCTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorLTOp<typename TensorUtils<THCTensor>::DataType,
                    typename TensorUtils<THCTensor>::DataType>());
}

THC_API void
THCTensor_(gtTensorT)(THCState *state, THCTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorGTOp<typename TensorUtils<THCTensor>::DataType,
                    typename TensorUtils<THCTensor>::DataType>());
}

THC_API void
THCTensor_(leTensorT)(THCState *state, THCTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorLEOp<typename TensorUtils<THCTensor>::DataType,
                    typename TensorUtils<THCTensor>::DataType>());
}

THC_API void
THCTensor_(geTensorT)(THCState *state, THCTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorGEOp<typename TensorUtils<THCTensor>::DataType,
                    typename TensorUtils<THCTensor>::DataType>());
}

THC_API void
THCTensor_(eqTensorT)(THCState *state, THCTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorEQOp<typename TensorUtils<THCTensor>::DataType,
                    typename TensorUtils<THCTensor>::DataType>());
}

THC_API void
THCTensor_(neTensorT)(THCState *state, THCTensor *self_, THCTensor *src1, THCTensor *src2)
{
  THAssert(THCTensor_(checkGPU)(state, 3, self_, src1, src2));
  THC_logicalTensor(state, self_, src1, src2,
                    TensorNEOp<typename TensorUtils<THCTensor>::DataType,
                    typename TensorUtils<THCTensor>::DataType>());
}

#endif