Home
last modified time | relevance | path

Searched refs:num_regular_fanins (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc100 const int num_regular_fanins = in HasRegularFaninNode() local
102 for (int i = 0; i < num_regular_fanins; ++i) { in HasRegularFaninNode()
847 int num_regular_fanins = in AddFaninInternal() local
855 can_dedup_control_with_regular_input ? 0 : num_regular_fanins; in AddFaninInternal()
865 input.port_id = input_is_control ? Graph::kControlSlot : num_regular_fanins; in AddFaninInternal()
872 if (num_regular_fanins < last_node_input) { in AddFaninInternal()
873 node->mutable_input()->SwapElements(last_node_input, num_regular_fanins); in AddFaninInternal()
884 max_regular_input_port()[node] = num_regular_fanins; in AddFaninInternal()
925 const int num_regular_fanins = in AddRegularFaninByPort() local
928 CheckPortRange(port, /*min=*/0, num_regular_fanins, error_status)); in AddRegularFaninByPort()
[all …]
Dgraph_view.h188 const int num_regular_fanins = in HasFanin() local
192 end = num_regular_fanins; in HasFanin()
194 start = num_regular_fanins; in HasFanin()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view_internal.h496 const int num_regular_fanins = node_view->NumRegularFanins(); in AddOrUpdateRegularFanin() local
497 if (index < num_regular_fanins) { // Updating existing fanins. in AddOrUpdateRegularFanin()
500 const int relative_removal_index = num_regular_fanins - index - 1; in AddOrUpdateRegularFanin()
519 const int relative_add_index = index - num_regular_fanins; in AddOrUpdateRegularFanin()
538 const int num_regular_fanins = node_view->NumRegularFanins(); in RemoveRegularFanin() local
539 if (index < num_regular_fanins) { // Removing existing fanins. in RemoveRegularFanin()
544 const int relative_removal_index = num_regular_fanins - index - 1; in RemoveRegularFanin()
552 const int relative_add_index = index - num_regular_fanins; in RemoveRegularFanin()
789 int num_regular_fanins = 0; member
834 ++new_node->num_regular_fanins; in AddOrUpdateRegularFanin()
[all …]
Dgraph_view.cc258 mutation_node.num_regular_fanins = mutation_node.regular_fanins.size(); in AddNode()
1146 const int num_regular_fanins = node_view->NumRegularFanins(); in RemoveControllingFaninInternal() local
1165 num_regular_fanins + control_index, in RemoveControllingFaninInternal()
1166 num_regular_fanins + node_view->NumControllingFanins() - 1); in RemoveControllingFaninInternal()
1280 new_node_def->mutable_input()->Reserve(new_node->num_regular_fanins + in SetNewNodesFanins()
1295 const int num_regular_fanins = node_view->NumRegularFanins(); in RemoveAllFaninFanoutInternal() local
1296 for (int i = 0; i < num_regular_fanins; ++i) { in RemoveAllFaninFanoutInternal()
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h138 GraphDef CreateFaninFanoutNodeGraph(int num_regular_fanins, in CreateFaninFanoutNodeGraph() argument
153 for (int i = 0; i < num_regular_fanins; ++i) { in CreateFaninFanoutNodeGraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc110 const int num_regular_fanins = node.NumRegularFanins(); in GetRegularFaninPorts() local
111 std::vector<int> values(num_regular_fanins); in GetRegularFaninPorts()
1138 const int num_regular_fanins = node.NumRegularFanins(); in GetVariadicNDFaninPorts() local
1139 ports.reserve(num_regular_fanins); in GetVariadicNDFaninPorts()
1140 for (int i = 0; i < num_regular_fanins; ++i) { in GetVariadicNDFaninPorts()