Searched refs:GetOpType (Results 1 – 11 of 11) sorted by relevance
25 auto it = cus_input_map_->find(op->GetOpType()); in IsCustomOp()126 auto it = cus_input_map_->find(op->GetOpType()); in SetCustomOpInput()162 auto it = cus_input_map_->find(op->GetOpType()); in SetCustomOpInput()246 auto it = cus_output_map_->find(op->GetOpType()); in getCustomOutput()289 if (cus_output_map_->find(op->GetOpType()) == cus_output_map_->end() || in UpdateSingleOutputDesc()290 ((*cus_output_map_)[op->GetOpType()].empty())) { in UpdateSingleOutputDesc()296 std::unordered_map<int, std::string> output_map = (*cus_output_map_)[op->GetOpType()]; in UpdateSingleOutputDesc()310 if (cus_output_map_->find(cus_op->GetOpType()) == cus_output_map_->end()) { in GetCustomOpOutputSize()314 size_t output_size = (*cus_output_map_)[cus_op->GetOpType()].size(); in GetCustomOpOutputSize()368 …:dynamic_pointer_cast<CustomOperator>(op)->UpdateOutputDesc((*cus_output_map_)[op->GetOpType()][i], in UpdateMultiOutputDesc()[all …]
611 …MS_LOG(INFO) << "op name: " << op->GetName() << ", op type: " << op->GetOpType() << ", out_name: "… in TraceOutputFromTupleGetItem()695 …MS_LOG(INFO) << "op name: " << op->GetName() << ", op type: " << op->GetOpType() << ", out_name: "… in TraceOutputFromParameter()701 MS_LOG(INFO) << "op name: " << op->GetName() << ", op type: " << op->GetOpType(); in TraceOutputFromParameter()1281 (op_itor->second->GetOpType() == "Constant" || op_itor->second->GetOpType() == "Const") && in GetRealInputNode()1347 if (op->GetOpType() == "Constant" || op->GetOpType() == "Const") { in AddGraphConstInput()
78 GetOpType(kernel_node); in Init()125 void GetOpType(const CNodePtr &kernel_node) { in GetOpType() function
60 GetOpType(kernel_node); in Init()119 void GetOpType(const CNodePtr &kernel_node) { in GetOpType() function
72 GetOpType(kernel_node); in Init()156 void GetOpType(const CNodePtr &kernel_node) { in GetOpType() function
73 GetOpType(kernel_node); in Init()154 void GetOpType(const CNodePtr &kernel_node) { in GetOpType() function
75 const std::string &GetOpType() const { return name_; } in GetOpType() function
76 const std::string &GetOpType() const { return name_; } in GetOpType() function
65 inline schema::PrimitiveType GetOpType(const schema::CNode &opDef) { return opDef.primitive()->valu… in GetOpType() function67 …tOpTypeName(const schema::CNode &opDef) { return schema::EnumNamePrimitiveType(GetOpType(opDef)); } in GetOpTypeName()
332 AstSubType GetOpType(const py::object &node);
898 AstSubType op_type = ast_->GetOpType(op_node); in ParseBoolOp()2067 AstSubType ParseFunctionAst::GetOpType(const py::object &node) { in GetOpType() function in mindspore::parse::ParseFunctionAst