Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator.cc67 int input_port; in AddCostNode() local
68 string input_name = ParseNodeName(input, &input_port); in AddCostNode()
88 input_info->set_preceding_port(input_port); in AddCostNode()
Dvirtual_scheduler.cc819 for (const auto& input_port : node_state.inputs) { in MarkCurrNodeExecuted() local
820 auto* input = input_port.first; in MarkCurrNodeExecuted()
821 auto port = input_port.second; in MarkCurrNodeExecuted()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_utils.cc231 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), 0); in GetInputNode() local
232 return graph.GetRegularFanin(input_port).node; in GetInputNode()
238 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), i); in GetInputNode() local
239 return graph.GetRegularFanin(input_port).node; in GetInputNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc240 const auto input_port = GraphView::InputPort(bias_add, 0); in FindConv2DWithBias() local
241 const auto conv2d = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBias()
271 const auto input_port = GraphView::InputPort(relu, 0); in FindConv2DWithBiasAndRelu() local
272 const auto bias_add = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBiasAndRelu()
368 const auto input_port = GraphView::InputPort(batch_norm, 0); in FindConv2DWithBatchNorm() local
369 const auto conv2d = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBatchNorm()
400 const auto input_port = GraphView::InputPort(node, 0); in FindConv2DWithBatchNormAndRelu() local
401 const auto batch_norm = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBatchNormAndRelu()
Dlayout_optimizer.cc672 int input_port; in AddTransformToOutputs() local
673 string input_name = ParseNodeName(input, &input_port); in AddTransformToOutputs()
677 if (output_pos.find(input_port) != output_pos.end()) { in AddTransformToOutputs()
683 graph_properties_.GetOutputProperties(node_->name())[input_port] in AddTransformToOutputs()
691 node_->attr().at("_output_shapes").list().shape(input_port), in AddTransformToOutputs()
1827 int input_port; in IsInputConvertible() local
1829 ParseNodeName(node_->input(0), &input_port); in IsInputConvertible()
1831 auto shape = input->attr().at("_output_shapes").list().shape(input_port); in IsInputConvertible()
Darithmetic_optimizer.cc1543 for (int input_port = start; input_port < end; ++input_port) { in InitializeChains() local
1544 if (IsControlInput(node.input(input_port))) { in InitializeChains()
1546 "Got control input ", node.input(input_port), in InitializeChains()
1550 TF_RETURN_IF_ERROR(GetInputNode(node.input(input_port), &tail)); in InitializeChains()
1551 tails->insert(ChainLink(tail, input_port)); in InitializeChains()
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc755 const InputPort& input_port) { in UpdateFanoutsInternal() argument
756 fanouts()[output_port].insert(input_port); in UpdateFanoutsInternal()
761 const InputPort& input_port) { in UpdateFanoutsInternal() argument
762 fanouts()[output_port].erase(input_port); in UpdateFanoutsInternal()
803 const InputPort input_port = edge.dst; in UpdateFanoutsInternal() local
807 if (input_port.node == to_node) { in UpdateFanoutsInternal()
814 input_port.node->set_input( in UpdateFanoutsInternal()
815 input_port.port_id, in UpdateFanoutsInternal()
819 remove_edge(output_port, input_port); in UpdateFanoutsInternal()
821 add_edge(OutputPort(to_node, output_port.port_id), input_port); in UpdateFanoutsInternal()
[all …]
Dmutable_graph_view_test.cc48 MutableGraphView::InputPort input_port(node, port); in CompareNodeFanins() local
51 EXPECT_TRUE(graph.GetFanin(input_port).contains(output_port)); in CompareNodeFanins()
52 EXPECT_TRUE(graph.GetFanout(output_port).contains(input_port)); in CompareNodeFanins()
63 MutableGraphView::InputPort input_port(graph.GetNode(tensor_id.node()), in CompareNodeFanouts() local
65 EXPECT_TRUE(node_fanouts.contains(input_port)); in CompareNodeFanouts()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc403 int input_port, DataType* input_type) { in InputTypeForNode() argument
407 if (input_types.size() > input_port) { in InputTypeForNode()
408 const DataType dtype = input_types[input_port]; in InputTypeForNode()
413 return errors::InvalidArgument("Input ", input_port, " not found for node ", in InputTypeForNode()
Dnode_def_util.h258 int input_port, DataType* input_type);
/external/kernel-headers/original/uapi/linux/
Dcec-funcs.h1899 __u8 input_port, in cec_msg_cdc_hpd_set_state() argument
1907 msg->msg[5] = (input_port << 4) | hpd_state; in cec_msg_cdc_hpd_set_state()
1912 __u8 *input_port, in cec_ops_cdc_hpd_set_state() argument
1916 *input_port = msg->msg[5] >> 4; in cec_ops_cdc_hpd_set_state()