Searched refs:GpuKernelArgument (Results 1 – 6 of 6) sorted by relevance
72 class GpuKernelArgument80 GpuKernelArgument(const ITensorInfo &tensor_info, in GpuKernelArgument() function
60 virtual std::map<ITensorInfo::Id, GpuKernelArgument> get_tensors() = 0;
41 using GpuKernelArgumentList = std::map<ITensorInfo::Id, GpuKernelArgument>;
121 std::map<ITensorInfo::Id, GpuKernelArgument> ClTemplateWriter::get_tensors() in get_tensors()124 std::map<ITensorInfo::Id, GpuKernelArgument> tensors; in get_tensors()129 GpuKernelArgument{ *t, _vtable.get_variable(t).kernel_argument_info }); in get_tensors()
76 std::map<ITensorInfo::Id, GpuKernelArgument> get_tensors() override;
510 "src/dynamic_fusion/sketch/gpu/GpuKernelArgument.cpp",