Searched refs:GetNumCommonOuterIterator (Results 1 – 5 of 5) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/cpp/ |
H A D | auto_scheduler_test.cc | 113 CHECK_EQ(dag->access_analyzer.GetNumCommonOuterIterator(dag->ops[conv], dag->ops[bias_add]), 4); in TEST() 114 CHECK_EQ(dag->access_analyzer.GetNumCommonOuterIterator(dag->ops[conv], dag->ops[relu]), 4); in TEST() 115 CHECK_EQ(dag->access_analyzer.GetNumCommonOuterIterator(dag->ops[data], dag->ops[relu]), 1); in TEST()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/auto_scheduler/ |
H A D | compute_dag.h | 155 TVM_DLL int GetNumCommonOuterIterator(const te::Operation& op,
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/search_policy/ |
H A D | utils.h | 268 inline int GetNumCommonOuterIterator(const SearchTask& task, const State& state, int stage_id, in GetNumCommonOuterIterator() function 272 ->access_analyzer.GetNumCommonOuterIterator(state->stages[stage_id]->op, in GetNumCommonOuterIterator() 275 return task->compute_dag->access_analyzer.GetNumCommonOuterIterator( in GetNumCommonOuterIterator()
|
H A D | sketch_policy_rules.cc | 364 GetNumCommonOuterIterator(policy.search_task, tmp_s, stage_id, target_stage_id); in Apply()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/auto_scheduler/ |
H A D | compute_dag.cc | 475 int AccessAnalyzer::GetNumCommonOuterIterator(const te::Operation& op, in GetNumCommonOuterIterator() function in tvm::auto_scheduler::AccessAnalyzer
|