Home
last modified time | relevance | path

Searched refs:branch_type (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc29 STATUS FunctionalizeCond::GetSwitchBranchType(const CNodePtr &switch_cnode, BranchType *branch_type… in GetSwitchBranchType() argument
31 MS_ASSERT(branch_type != nullptr); in GetSwitchBranchType()
51 *branch_type = kElseBranch; in GetSwitchBranchType()
53 *branch_type = kThenBranch; in GetSwitchBranchType()
62 BranchType branch_type) { in BranchSubGraphAddNodes() argument
76 if (GetSwitchBranchType(cnode, &this_type) != RET_OK || this_type != branch_type) { in BranchSubGraphAddNodes()
77 MS_LOG(ERROR) << "switch node in branch " << branch_type << " is not correct"; in BranchSubGraphAddNodes()
147 …ionalizeCond::CreateBranchGraph(const AnfNodePtr &node, std::string name, BranchType branch_type) { in CreateBranchGraph() argument
154 auto status = BranchSubGraphAddNodes(graph, node, branch_type); in CreateBranchGraph()
Dfunctionalize_cond.h44 STATUS GetSwitchBranchType(const CNodePtr &switch_cnode, BranchType *branch_type);
45 …chSubGraphAddNodes(const FuncGraphPtr &graph, const AnfNodePtr &root_node, BranchType branch_type);
46 FuncGraphPtr CreateBranchGraph(const AnfNodePtr &node, std::string name, BranchType branch_type);
/third_party/mesa3d/src/compiler/spirv/
Dvtn_cfg.c689 block->branch_type = vtn_handle_branch(b, cf_parent, branch_block); in vtn_process_block()
691 if (block->branch_type == vtn_branch_type_none) in vtn_process_block()
699 block->branch_type = vtn_branch_type_return; in vtn_process_block()
703 block->branch_type = vtn_branch_type_discard; in vtn_process_block()
707 block->branch_type = vtn_branch_type_terminate_invocation; in vtn_process_block()
711 block->branch_type = vtn_branch_type_ignore_intersection; in vtn_process_block()
715 block->branch_type = vtn_branch_type_terminate_ray; in vtn_process_block()
937 vtn_emit_branch(struct vtn_builder *b, enum vtn_branch_type branch_type, in vtn_emit_branch() argument
940 switch (branch_type) { in vtn_emit_branch()
1082 if (block->branch_type != vtn_branch_type_none) { in vtn_emit_cf_list_structured()
[all …]
Dvtn_private.h239 enum vtn_branch_type branch_type; member