Home
last modified time | relevance | path

Searched refs:merge_node (Results 1 – 11 of 11) sorted by relevance

/external/mesa3d/src/compiler/nir/
Dnir_from_ssa.c104 } merge_node; typedef
119 foreach_list_typed(merge_node, node, node, &set->nodes) {
136 static merge_node *
149 merge_node *node = ralloc(state->dead_ctx, merge_node); in get_merge_node()
160 merge_nodes_interfere(merge_node *a, merge_node *b) in merge_nodes_interfere()
172 merge_node *a_node = exec_node_data(merge_node, an, node); in merge_merge_sets()
173 merge_node *b_node = exec_node_data(merge_node, bn, node); in merge_merge_sets()
180 exec_node_data(merge_node, bn, node)->set = a; in merge_merge_sets()
202 NIR_VLA(merge_node *, dom, a->size + b->size); in merge_sets_interfere()
210 merge_node *current; in merge_sets_interfere()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op.cc282 Node* merge_node; in CreateMergeNodes() local
289 .Finalize(graph_, &merge_node)); in CreateMergeNodes()
290 merge_nodes_.emplace_back(merge_node); in CreateMergeNodes()
329 Node* merge_node = merge_nodes_[op_input_output_to_lowered_node_[i]]; in CreateSwitchNodes() local
332 if (IsRefType(merge_node->output_type(0))) { in CreateSwitchNodes()
337 .Input(NodeOut(merge_node, 0)) in CreateSwitchNodes()
339 .Device(merge_node->requested_device()) in CreateSwitchNodes()
340 .AssignedDevice(merge_node->assigned_device_name()) in CreateSwitchNodes()
447 Node* merge_node = merge_nodes_[op_input_output_to_lowered_node_[i]]; in CreateNextIterationNodes() local
452 .Device(merge_node->requested_device()) in CreateNextIterationNodes()
[all …]
Dlower_while_op_test.cc227 const Node* merge_node = nullptr; in TEST() local
231 merge_node = consumer; in TEST()
238 for (const Node* consumer : merge_node->in_nodes()) { in TEST()
248 for (const Node* consumer : merge_node->out_nodes()) { in TEST()
348 const Node* merge_node = nullptr; in TEST() local
352 merge_node = consumer; in TEST()
359 for (const Node* consumer : merge_node->in_nodes()) { in TEST()
369 for (const Node* consumer : merge_node->out_nodes()) { in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc641 NodeDef* merge_node = nullptr; in CheckForDeadFanout() local
650 merge_node = node; in CheckForDeadFanout()
657 if (merge_node == nullptr || constant_ctrl_input == nullptr) { in CheckForDeadFanout()
664 for (const auto& input : merge_node->input()) { in CheckForDeadFanout()
689 << "\n| merge_node : " << merge_node->name() in CheckForDeadFanout()
897 NodeDef* merge_node = itr.first; in RemoveDeadBranches() local
898 if (dead_nodes.find(merge_node) != dead_nodes.end()) { in RemoveDeadBranches()
904 const int num_data_inputs = merge_node->attr().at("N").i(); in RemoveDeadBranches()
905 if (merge_node->input_size() != num_data_inputs) { in RemoveDeadBranches()
908 << merge_node->name(); in RemoveDeadBranches()
[all …]
Dlayout_optimizer_test.cc1140 auto merge_node = node_map.GetNode("merge"); in TEST_F() local
1141 EXPECT_EQ(merge_node->input(0), "Conv2D"); in TEST_F()
1142 EXPECT_EQ(merge_node->input(1), "i1"); in TEST_F()
1162 auto merge_node = node_map.GetNode("merge"); in TEST_F() local
1163 EXPECT_EQ(merge_node->input(0), "tensor_4d"); in TEST_F()
1164 EXPECT_EQ(merge_node->input(1), in TEST_F()
1179 auto merge_node = node_map.GetNode("complex"); in TEST_F() local
1180 EXPECT_EQ(merge_node->input(0), "Conv2D"); in TEST_F()
1181 EXPECT_EQ(merge_node->input(1), "Conv2D"); in TEST_F()
1224 auto merge_node = node_map_original.GetNode("merge"); in TEST_F() local
[all …]
Dauto_mixed_precision.cc1737 const NodeDef& merge_node = *output.node; in ForceColorMatchOnRecurrentEdges() local
1738 if (merge_node.op() != "Merge") { in ForceColorMatchOnRecurrentEdges()
1740 "Expected Merge node after NextIteration, got ", merge_node.op()); in ForceColorMatchOnRecurrentEdges()
1743 graph_type_view_.GetNodeIndex(merge_node.name(), TypeAttrId("T")); in ForceColorMatchOnRecurrentEdges()
1746 merge_node.name(), in ForceColorMatchOnRecurrentEdges()
Dauto_mixed_precision_test.cc658 auto merge_node = node_map_original.GetNode("mrg1"); in TEST_F() local
660 merge_node->set_input(1, "nxt1"); in TEST_F()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dstruct_cfg_analysis.cpp47 uint32_t merge_node; in AddBlocksInFunction() member
58 state[0].merge_node = 0; in AddBlocksInFunction()
67 if (block->id() == state.back().merge_node) { in AddBlocksInFunction()
81 new_state.merge_node = in AddBlocksInFunction()
110 merge_blocks_.Set(new_state.merge_node); in AddBlocksInFunction()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dstruct_cfg_analysis.cpp47 uint32_t merge_node; in AddBlocksInFunction() member
58 state[0].merge_node = 0; in AddBlocksInFunction()
67 if (block->id() == state.back().merge_node) { in AddBlocksInFunction()
81 new_state.merge_node = in AddBlocksInFunction()
110 merge_blocks_.Set(new_state.merge_node); in AddBlocksInFunction()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dstruct_cfg_analysis.cpp47 uint32_t merge_node; in AddBlocksInFunction() member
58 state[0].merge_node = 0; in AddBlocksInFunction()
67 if (block->id() == state.back().merge_node) { in AddBlocksInFunction()
81 new_state.merge_node = in AddBlocksInFunction()
110 merge_blocks_.Set(new_state.merge_node); in AddBlocksInFunction()
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop.cc83 Node* merge_node; in CreateMerge() local
84 TF_RETURN_IF_ERROR(builder.Finalize(scope.graph(), &merge_node)); in CreateMerge()
85 TF_RETURN_IF_ERROR(scope.DoShapeInference(merge_node)); in CreateMerge()
86 *merge_output = Output(merge_node, 0); in CreateMerge()