Searched defs:BinaryFunctor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cwise_op_gpu_logical_and.cu.cc | 23 template struct BinaryFunctor<GPUDevice, logical_and, 1>; variable
24 template struct BinaryFunctor<GPUDevice, logical_and, 2>; variable
25 template struct BinaryFunctor<GPUDevice, logical_and, 3>; variable
26 template struct BinaryFunctor<GPUDevice, logical_and, 4>; variable
27 template struct BinaryFunctor<GPUDevice, logical_and, 5>; variable
|
| H A D | cwise_op_gpu_logical_or.cu.cc | 23 template struct BinaryFunctor<GPUDevice, logical_or, 1>; variable
24 template struct BinaryFunctor<GPUDevice, logical_or, 2>; variable
25 template struct BinaryFunctor<GPUDevice, logical_or, 3>; variable
26 template struct BinaryFunctor<GPUDevice, logical_or, 4>; variable
27 template struct BinaryFunctor<GPUDevice, logical_or, 5>; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Loops.cuh | 152 struct BinaryFunctor { struct
153 __device__ return_t operator()(arg1_t a, arg2_t b) const { in operator ()()
156 BinaryFunctor(func_t f_): f(f_) {} in BinaryFunctor() argument
158 func_t f;
|