Searched defs:src_num (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/examples/runtime_gpu_extend/src/ |
| D | custom_common.h | 33 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num) { in Broadcast2GpuShape()
61 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-src/source/mindspore/lite/src/litert/kernel/opencl/ |
| D | opencl_kernel.h | 61 int Broadcast2GpuShape(const SrcT *src, int src_num, DstT *dst, int dsc_num) { in Broadcast2GpuShape()
96 int Broadcast2GpuShape(const SrcT *src, int src_num, DstT *dst, int dsc_num, DstT default_value) { in Broadcast2GpuShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/registry/ |
| D | registry_gpu_custom_op_test.cc | 66 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num) { in Broadcast2GpuShape()
94 void Broadcast2GpuShape(DstT *dst, const SrcT *src, int src_num, DstT default_value) { in Broadcast2GpuShape()
|