Searched defs:input_addr1 (Results 1 – 4 of 4) 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 | 131 T *input_addr1 = GetDeviceAddress<T>(inputs, 0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | glu_fp32.cc | 134 auto input_addr1 = reinterpret_cast<float *>(sigmoid_ptr_); in Mul() local
|