Home
last modified time | relevance | path

Searched defs:kernel_input (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.cc271 session::KernelWithIndex kernel_input; in GetKernelInputRef() local
420 session::KernelWithIndex kernel_input; in SetGraphOutputRefCount() local
Dmem_reuse_checker.cc219 auto kernel_input = kernel_input_with_idx.first; in CheckGraphOutputAssigned() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc301 auto kernel_input = AnfAlgo::VisitKernelWithReturnType(input_node, 0, true); in GetCommunicationOpInputInfo() local
/third_party/mesa3d/src/asahi/compiler/
Dagx_compile.c323 bool kernel_input = (instr->intrinsic == nir_intrinsic_load_kernel_input); in agx_emit_load_ubo() local
/third_party/mesa3d/src/panfrost/bifrost/
Dbifrost_compile.c732 bool kernel_input = (instr->intrinsic == nir_intrinsic_load_kernel_input); in bi_emit_load_ubo() local