Home
last modified time | relevance | path

Searched refs:output_port (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc257 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape() argument
264 if (output_port < 0 || output_port >= node->num_outputs()) { in SetShape()
266 "output_port '", output_port, "' is out of range, ", "node '", in SetShape()
276 ShapeHandle existing_shape = c->output(output_port); in SetShape()
278 c->set_output(output_port, shape); in SetShape()
Dshape_refiner.h99 Status SetShape(const Node* node, int output_port,
/external/tensorflow/tensorflow/core/framework/
Dgraph_transfer_info.proto15 int32 output_port = 2; field
Dnode_def_util.cc536 int output_port, DataType* output_type) { in OutputTypeForNode() argument
541 if (output_types_size > output_port) { in OutputTypeForNode()
542 const DataType dtype = output_types[output_port]; in OutputTypeForNode()
547 return errors::InvalidArgument("Output ", output_port, " not found for node ", in OutputTypeForNode()
Dnode_def_util.h331 int output_port, DataType* output_type);
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc757 const auto add_edge = [this](const OutputPort& output_port, in UpdateFanoutsInternal()
759 fanouts()[output_port].insert(input_port); in UpdateFanoutsInternal()
763 const auto remove_edge = [this](const OutputPort& output_port, in UpdateFanoutsInternal() argument
765 fanouts()[output_port].erase(input_port); in UpdateFanoutsInternal()
805 const OutputPort output_port = edge.src; in UpdateFanoutsInternal() local
812 std::max(keep_max_regular_output_port, output_port.port_id); in UpdateFanoutsInternal()
819 TensorIdToString({to_node->name(), output_port.port_id})); in UpdateFanoutsInternal()
822 remove_edge(output_port, input_port); in UpdateFanoutsInternal()
824 add_edge(OutputPort(to_node, output_port.port_id), input_port); in UpdateFanoutsInternal()
Dmutable_graph_view_test.cc49 MutableGraphView::OutputPort output_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()
1186 MutableGraphView::OutputPort output_port = in CheckFanoutRemoved() local
1188 auto fanouts = graph.GetFanout(output_port); in CheckFanoutRemoved()
/external/tensorflow/tensorflow/python/grappler/
Dauto_mixed_precision_test.py353 def _assert_output_f16(self, mode, node_map, node_name, output_port=0): argument
354 self.assertEqual(node_map[node_name].output_info[output_port].dtype,
422 output_port = 0
423 optimized_type = node_map[node_name].output_info[output_port].dtype
/external/tcpdump/
Dprint-openflow-1.0.c1451 uint16_t type, alen, output_port; in of10_actions_print() local
1510 output_port = EXTRACT_16BITS(cp); in of10_actions_print()
1512 ND_PRINT((ndo, ", port %s", tok2str(ofpp_str, "%u", output_port))); in of10_actions_print()
1515 if (output_port == OFPP_CONTROLLER) in of10_actions_print()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc1731 GraphView::OutputPort output_port(&node, 0); in ForceColorMatchOnRecurrentEdges() local
1732 const auto& fanout = graph_view_.GetFanout(output_port); in ForceColorMatchOnRecurrentEdges()
1918 for (int output_port : node_type_map_.GetOutputPorts(*node, type_attr)) { in ChangeTypeAttrsAndAddCasts() local
1919 MutableGraphView::OutputPort src(node, output_port); in ChangeTypeAttrsAndAddCasts()
Dlayout_optimizer.cc381 bool IsHostMemory(const NodeDef& node, int output_port) { in IsHostMemory() argument
392 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()
Dgeneric_layout_optimizer_transposer.cc139 bool IsHostMemory(const NodeDef& node, int output_port) { in IsHostMemory() argument
150 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1100 Status SetUnknownShape(const NodeDef* node, int output_port) { in SetUnknownShape() argument
1102 GetUnknownOutputShape(node, output_port); in SetUnknownShape()
1107 ctx->set_output(output_port, shape); in SetUnknownShape()
1897 for (const int output_port : it->second) { in InferShapes() local
1898 status.Update(SetUnknownShape(&node, output_port)); in InferShapes()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h413 Status ConvertToTensorOrWeights(const NodeDef& node_def, int output_port,
Dconvert_nodes.cc1148 const NodeDef& node_def, int output_port, in ConvertToTensorOrWeights() argument
1151 if (output_port != 0) { in ConvertToTensorOrWeights()
1171 const auto& tensor_properties = output_params.at(output_port); in ConvertToTensorOrWeights()
Dconvert_nodes_test.cc535 int output_port, in ConvertToTensorOrWeights() argument
545 return validator.ConvertToTensorOrWeights(node->def(), output_port, in ConvertToTensorOrWeights()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc217 ports[i] = node_input.output_port(); in SetupGraph()
Dgraph_transferer_test.cc249 EXPECT_GE(node_input.output_port(), 0); in SanityCheckNodes()
Dgraph_transferer.cc1157 << ", output port = " << node_input.output_port(); in DumpNodeTransferParams()
1210 << node_input.output_port(); in DumpVerificationStringOfNodeTransferParams()