Home
last modified time | relevance | path

Searched refs:is_control (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc51 int dst_port, bool is_control = false) in SimpleEdge() argument
57 control_(is_control) {} in SimpleEdge()
168 bool is_control = e->IsControlEdge(); in SimpleGraph() local
172 is_control); in SimpleGraph()
192 bool is_control = (out_port == Graph::kControlSlot); in AddEdge() local
193 is_control |= (in_port == Graph::kControlSlot); in AddEdge()
194 auto edge = new SimpleEdge(i, src, out_port, dst, in_port, is_control); in AddEdge()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc52 const auto is_control = [](const string& input) -> bool { in CompareGraphNodes() local
58 std::sort(absl::c_find_if(*want_inputs, is_control), want_inputs->end()); in CompareGraphNodes()
59 std::sort(absl::c_find_if(*got_inputs, is_control), got_inputs->end()); in CompareGraphNodes()
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py817 self.assertEqual([], dump.node_inputs(u_name, is_control=True))
818 self.assertEqual([], dump.node_inputs(u_read_name, is_control=True))
819 self.assertEqual([], dump.node_inputs(v_name, is_control=True))
820 self.assertEqual([], dump.node_inputs(w_name, is_control=True))
827 self.assertEqual([], dump.node_recipients(u_name, is_control=True))
828 self.assertEqual([], dump.node_recipients(u_read_name, is_control=True))
829 self.assertEqual([], dump.node_recipients(v_name, is_control=True))
830 self.assertEqual([], dump.node_recipients(w_name, is_control=True))
Ddebug_data.py1067 def node_inputs(self, node_name, is_control=False, device_name=None): argument
1089 if is_control:
1230 def node_recipients(self, node_name, is_control=False, device_name=None): argument
1254 if is_control:
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli.py803 self._debug_dump.node_inputs(node_name, is_control=True))
810 self._debug_dump.node_recipients(node_name, is_control=True))
1407 copy.copy(tracker(node_name, is_control=False)))
1413 sorted(tracker(node_name, is_control=True)))
/external/tensorflow/tensorflow/core/summary/
Dsummary_db_writer.cc266 int64 is_control = 0; in SaveNodeInputs() local
277 is_control = 1; in SaveNodeInputs()
289 insert.BindInt(6, is_control); in SaveNodeInputs()
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc1592 bool is_control = IsTensorIdControlling(tensor_id); in RemoveFaninsInternal() local
1593 if (keep_controlling_fanins && is_control) { in RemoveFaninsInternal()
1600 input.port_id = is_control ? Graph::kControlSlot : i; in RemoveFaninsInternal()