Home
last modified time | relevance | path

Searched defs:src_num (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/lite/examples/runtime_gpu_extend/src/
Dcustom_common.h35 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/
Dir3_lower_parallelcopy.c151 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/
Dopencl_kernel.h52 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/
Dregistry_gpu_custom_op_test.cc68 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()