Searched defs:input_addr2 (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | logical_impl.cu | 34 __global__ void LogicalAndKernel(const T *input_addr1, const T *input_addr2, T *output, const int s… in LogicalAndKernel()
41 __global__ void LogicalOrKernel(const T *input_addr1, const T *input_addr2, T *output, const int si… in LogicalOrKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/cuda_impl/ |
| D | logical.cu | 28 __global__ void LogicalAndKernel(const T *input_addr1, const T *input_addr2, T *output, int size) { in LogicalAndKernel()
35 __global__ void LogicalOrKernel(const T *input_addr1, const T *input_addr2, T *output, int size) { in LogicalOrKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | addv2_gpu_kernel.cc | 132 T *input_addr2 = GetDeviceAddress<T>(inputs, 1); in LaunchKernel() local
|