| /dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/ |
| H A D | ir_pass.h | 446 LoweredFunc BindDeviceType(LoweredFunc func,
463 Array<LoweredFunc> SplitHostDevice(LoweredFunc func);
479 LoweredFunc LowerThreadAllreduce(LoweredFunc f, int warp_size);
488 LoweredFunc LowerWarpMemory(LoweredFunc f, int warp_size);
509 LoweredFunc LowerTVMBuiltin(LoweredFunc f);
516 LoweredFunc CombineContextCall(LoweredFunc f);
528 LoweredFunc PointerValueTypeRewrite(LoweredFunc f);
537 LoweredFunc LowerDeviceStorageAccessInfo(LoweredFunc func);
545 LoweredFunc LowerIntrin(LoweredFunc f, const std::string& target);
564 LoweredFunc InferFragment(LoweredFunc f);
[all …]
|
| H A D | lowered_func.h | 44 class LoweredFunc : public ir::FunctionRef {
46 LoweredFunc() {} in LoweredFunc() function
47 explicit LoweredFunc(ObjectPtr<Object> n) : FunctionRef(n) {} in LoweredFunc() function
138 inline const LoweredFuncNode* LoweredFunc::operator->() const {
146 struct hash<::tvm::LoweredFunc> : public tvm::NodeHash {
|
| H A D | build_module.h | 309 TVM_DLL Array<LoweredFunc> lower(Schedule sch,
323 TVM_DLL Array<Array<LoweredFunc> > split_dev_host_funcs(const Array<LoweredFunc>& funcs,
336 TVM_DLL runtime::Module build(const Array<LoweredFunc>& funcs,
351 TVM_DLL runtime::Module build(const Map<Target, Array<LoweredFunc>>& input,
366 TVM_DLL runtime::Module build(const Map<std::string, Array<LoweredFunc>>& input,
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/ |
| H A D | ir_pass.h | 446 LoweredFunc BindDeviceType(LoweredFunc func,
463 Array<LoweredFunc> SplitHostDevice(LoweredFunc func);
479 LoweredFunc LowerThreadAllreduce(LoweredFunc f, int warp_size);
488 LoweredFunc LowerWarpMemory(LoweredFunc f, int warp_size);
509 LoweredFunc LowerTVMBuiltin(LoweredFunc f);
516 LoweredFunc CombineContextCall(LoweredFunc f);
528 LoweredFunc PointerValueTypeRewrite(LoweredFunc f);
537 LoweredFunc LowerDeviceStorageAccessInfo(LoweredFunc func);
545 LoweredFunc LowerIntrin(LoweredFunc f, const std::string& target);
564 LoweredFunc InferFragment(LoweredFunc f);
[all …]
|
| H A D | lowered_func.h | 44 class LoweredFunc : public ir::FunctionRef {
46 LoweredFunc() {} in LoweredFunc() function
47 explicit LoweredFunc(ObjectPtr<Object> n) : FunctionRef(n) {} in LoweredFunc() function
138 inline const LoweredFuncNode* LoweredFunc::operator->() const {
146 struct hash<::tvm::LoweredFunc> : public tvm::NodeHash {
|
| H A D | build_module.h | 309 TVM_DLL Array<LoweredFunc> lower(Schedule sch,
323 TVM_DLL Array<Array<LoweredFunc> > split_dev_host_funcs(const Array<LoweredFunc>& funcs,
336 TVM_DLL runtime::Module build(const Array<LoweredFunc>& funcs,
351 TVM_DLL runtime::Module build(const Map<Target, Array<LoweredFunc>>& input,
366 TVM_DLL runtime::Module build(const Map<std::string, Array<LoweredFunc>>& input,
|
| /dports/lang/halide/Halide-release_2019_08_27-2654-g664dc4993/src/ |
| H A D | Module.h | 97 struct LoweredFunc { struct
113 LoweredFunc(const std::string &name, argument
118 LoweredFunc(const std::string &name,
159 const std::vector<Internal::LoweredFunc> &functions() const;
160 std::vector<Internal::LoweredFunc> &functions();
167 Internal::LoweredFunc get_function_by_name(const std::string &name) const;
172 void append(const Internal::LoweredFunc &function);
|
| H A D | DebugArguments.h | 15 struct LoweredFunc;
19 void debug_arguments(LoweredFunc *func, const Target &t);
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/pass/ |
| H A D | split_host_device.cc | 181 Array<LoweredFunc> Split(LoweredFunc f) { in Split()
191 Array<LoweredFunc> ret{LoweredFunc(n)}; in Split()
192 for (LoweredFunc x : device_funcs_) { in Split()
226 LoweredFunc f_device(n); in SplitDeviceFunc()
244 std::vector<LoweredFunc> device_funcs_;
258 Array<LoweredFunc> SplitHostDevice(LoweredFunc func) { in SplitHostDevice()
|
| H A D | skip_assert.cc | 40 LoweredFunc SkipAssert(LoweredFunc f) { in SkipAssert()
43 return LoweredFunc(n); in SkipAssert()
|
| H A D | remap_thread_axis.cc | 78 LoweredFunc
79 RemapThreadAxis(LoweredFunc f, Map<Expr, IterVar> thread_map) { in RemapThreadAxis()
97 return LoweredFunc(n); in RemapThreadAxis()
|
| H A D | combine_context_call.cc | 110 LoweredFunc CombineContextCall(LoweredFunc f) { in CombineContextCall()
113 return LoweredFunc(n); in CombineContextCall()
|
| H A D | verify_memory.cc | 46 explicit MemoryAccessVerifier(LoweredFunc f, int device_type) in MemoryAccessVerifier()
176 LoweredFunc func_{nullptr}; ///< Function to be verified.
183 bool VerifyMemory(LoweredFunc func, int device_type) { in VerifyMemory()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/pass/ |
| H A D | split_host_device.cc | 181 Array<LoweredFunc> Split(LoweredFunc f) { in Split()
191 Array<LoweredFunc> ret{LoweredFunc(n)}; in Split()
192 for (LoweredFunc x : device_funcs_) { in Split()
226 LoweredFunc f_device(n); in SplitDeviceFunc()
244 std::vector<LoweredFunc> device_funcs_;
258 Array<LoweredFunc> SplitHostDevice(LoweredFunc func) { in SplitHostDevice()
|
| H A D | skip_assert.cc | 40 LoweredFunc SkipAssert(LoweredFunc f) { in SkipAssert()
43 return LoweredFunc(n); in SkipAssert()
|
| H A D | remap_thread_axis.cc | 78 LoweredFunc
79 RemapThreadAxis(LoweredFunc f, Map<Expr, IterVar> thread_map) { in RemapThreadAxis()
97 return LoweredFunc(n); in RemapThreadAxis()
|
| H A D | combine_context_call.cc | 110 LoweredFunc CombineContextCall(LoweredFunc f) { in CombineContextCall()
113 return LoweredFunc(n); in CombineContextCall()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/codegen/ |
| H A D | codegen_vhls.cc | 72 void CodeGenVivadoHLS::AddFunction(LoweredFunc f) { in AddFunction()
77 void CodeGenVivadoHLS::PreFunctionBody(LoweredFunc f) { in PreFunctionBody()
130 runtime::Module BuildSDAccel(Array<LoweredFunc> funcs, std::string target_str) { in BuildSDAccel()
137 for (LoweredFunc f : funcs) { in BuildSDAccel()
144 for (LoweredFunc f : funcs) { in BuildSDAccel()
|
| H A D | codegen_vhls.h | 39 void AddFunction(LoweredFunc f);
40 void PreFunctionBody(LoweredFunc f);
|
| H A D | build_common.h | 37 ExtractFuncInfo(const Array<LoweredFunc>& funcs) { in ExtractFuncInfo()
39 for (LoweredFunc f : funcs) { in ExtractFuncInfo()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/codegen/ |
| H A D | codegen_vhls.cc | 72 void CodeGenVivadoHLS::AddFunction(LoweredFunc f) { in AddFunction()
77 void CodeGenVivadoHLS::PreFunctionBody(LoweredFunc f) { in PreFunctionBody()
130 runtime::Module BuildSDAccel(Array<LoweredFunc> funcs, std::string target_str) { in BuildSDAccel()
137 for (LoweredFunc f : funcs) { in BuildSDAccel()
144 for (LoweredFunc f : funcs) { in BuildSDAccel()
|
| H A D | codegen_vhls.h | 39 void AddFunction(LoweredFunc f);
40 void PreFunctionBody(LoweredFunc f);
|
| H A D | build_common.h | 37 ExtractFuncInfo(const Array<LoweredFunc>& funcs) { in ExtractFuncInfo()
39 for (LoweredFunc f : funcs) { in ExtractFuncInfo()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/ |
| H A D | build_module.py | 63 if not isinstance(retv, (_stmt.Stmt, container.LoweredFunc, container.Array)):
68 out = retv.body if isinstance(retv, container.LoweredFunc) else retv
72 out = x.body if isinstance(x, container.LoweredFunc) else x
464 if func.func_type == container.LoweredFunc.MixedFunc:
476 elif func.func_type == container.LoweredFunc.HostFunc:
478 elif func.func_type == container.LoweredFunc.DeviceFunc:
591 if isinstance(flist, container.LoweredFunc):
593 elif isinstance(inputs, container.LoweredFunc):
617 if not isinstance(x, container.LoweredFunc):
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/ |
| H A D | build_module.py | 63 if not isinstance(retv, (_stmt.Stmt, container.LoweredFunc, container.Array)):
68 out = retv.body if isinstance(retv, container.LoweredFunc) else retv
72 out = x.body if isinstance(x, container.LoweredFunc) else x
464 if func.func_type == container.LoweredFunc.MixedFunc:
476 elif func.func_type == container.LoweredFunc.HostFunc:
478 elif func.func_type == container.LoweredFunc.DeviceFunc:
591 if isinstance(flist, container.LoweredFunc):
593 elif isinstance(inputs, container.LoweredFunc):
617 if not isinstance(x, container.LoweredFunc):
|