Searched refs:expand_info (Results 1 – 6 of 6) sorted by relevance
24 def create_expander(expand_info): argument28 op_name = str(expand_info['name'])32 return call_func(expander, expand_info)45 expand_info = {52 return expand_info59 expand_info = extract_expand_info(kernel_info)61 expander = create_expander(expand_info)
29 def __init__(self, expand_info): argument30 self.name = expand_info["name"]31 self.inputs = expand_info["input_desc"]32 self.outputs = expand_info["output_desc"]33 self.attrs = expand_info["attr"]34 self.processor = expand_info["process"]
24 def __init__(self, expand_info): argument25 super().__init__(expand_info)
27 def __init__(self, expand_info): argument28 super(MatMul, self).__init__(expand_info)
49 def __init__(self, expand_info): argument50 super().__init__(expand_info)
400 py::tuple expand_info = ast_->CallParseModFunction(PYTHON_PARSE_EXPAND_EXPR_STATEMENT, node); in ParseExpr() local408 if (expand_info.empty()) { in ParseExpr()411 auto is_expand = py::cast<bool>(expand_info[0]); in ParseExpr()415 if (expand_info.size() < expect_size) { in ParseExpr()416 …MS_LOG(EXCEPTION) << "expand_info size:" << expand_info.size() << " less than " << expect_size << … in ParseExpr()418 py::object value_object = expand_info[1]; in ParseExpr()421 if (py::len(expand_info) == 2) { in ParseExpr()435 py::object target_node = expand_info[2]; in ParseExpr()