Searched defs:src_num (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/lite/examples/runtime_gpu_extend/src/ |
| D | custom_common.h | 35 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num) { in Broadcast2GpuShape() 63 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num, DstT default_value) { in Broadcast2GpuShape()
|
| /third_party/mesa3d/src/freedreno/ir3/ |
| D | ir3_lower_parallelcopy.c | 151 unsigned src_num = ra_physreg_to_num(entry->src.reg, entry->flags); in do_swap() local 226 unsigned src_num = ra_physreg_to_num(entry->src.reg & ~1u, in do_copy() local 252 unsigned src_num = ra_physreg_to_num(entry->src.reg, entry->flags); in do_copy() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/ |
| D | opencl_kernel.h | 52 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num) { in Broadcast2GpuShape() 81 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num, DstT default_value) { in Broadcast2GpuShape()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/registry/ |
| D | registry_gpu_custom_op_test.cc | 68 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num) { in Broadcast2GpuShape() 96 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num, DstT default_value) { in Broadcast2GpuShape()
|