Searched refs:input_edges (Results 1 – 15 of 15) sorted by relevance
98 if (entry.input != node->input_edges().end()) { in RunUndirectedDFS()149 if (entry.input != node->input_edges().end()) { in RunUndirectedDFS()158 DCHECK(entry.input == node->input_edges().end()); in RunUndirectedDFS()193 Node::InputEdges::iterator input = node->input_edges().begin(); in DFSPush()
113 inline InputEdges input_edges();401 Node::InputEdges Node::input_edges() { in input_edges() function
169 for (Edge const edge : node->input_edges()) { in UpdatePlacement()1230 stack.push(node->input_edges().begin()); in PrepareUses()1236 if (++stack.top() == edge.from()->input_edges().end()) stack.pop(); in PrepareUses()1240 if (node->InputCount() > 0) stack.push(node->input_edges().begin()); in PrepareUses()
386 for (Edge edge : this->input_edges()) { in Verify()
177 TF_RETURN_IF_ERROR(n->input_edges(&input_edges_vector)); in GetXlaClusterInfo()178 absl::Span<const Edge*> input_edges(input_edges_vector); in GetXlaClusterInfo() local180 absl::c_transform(input_edges.subspan(0, num_constant_inputs), in GetXlaClusterInfo()185 input_edges.subspan(num_constant_inputs, num_non_constant_inputs), in GetXlaClusterInfo()189 input_edges.subspan(num_constant_inputs + num_non_constant_inputs, in GetXlaClusterInfo()
262 TF_RETURN_IF_ERROR(launch->input_edges(&in_edges)); in BuildXlaLaunchOps()
265 Status Node::input_edges(std::vector<const Edge*>* input_edges) const { in input_edges() function in tensorflow::Node266 input_edges->clear(); in input_edges()267 input_edges->resize(num_inputs(), nullptr); in input_edges()274 if ((*input_edges)[edge->dst_input()] != nullptr) { in input_edges()278 (*input_edges)[edge->dst_input()] = edge; in input_edges()282 if ((*input_edges)[i] == nullptr) { in input_edges()
251 TF_ASSERT_OK(t->input_edges(&t_input_edges)); in TEST_F()287 TF_ASSERT_OK(c->input_edges(&c_input_edges)); in TEST_F()588 EXPECT_EQ(error::INVALID_ARGUMENT, b->input_edges(&edges).code()); in TEST_F()590 TF_EXPECT_OK(b->input_edges(&edges)); in TEST_F()
214 TF_CHECK_OK(n->input_edges(&edges)); in InsertInputConversionNode()
195 Status input_edges(std::vector<const Edge*>* edges) const;
150 TF_RETURN_IF_ERROR(node->input_edges(in_edges)); in FindSaveOp()
249 std::vector<const Edge*> input_edges; in AddConversionMapping() local250 TF_RETURN_IF_ERROR(op_node->input_edges(&input_edges)); in AddConversionMapping()255 auto edge = input_edges[i]; in AddConversionMapping()
74 std::vector<const Edge*> input_edges; in IsTensorRTCandidate() local75 TF_RETURN_IF_ERROR(node->input_edges(&input_edges)); in IsTensorRTCandidate()77 input_node_and_ports.reserve(input_edges.size()); in IsTensorRTCandidate()78 for (const Edge* input_edge : input_edges) { in IsTensorRTCandidate()
174 TF_RETURN_IF_ERROR(if_op_->input_edges(&edges)); in AddInputs()
195 TF_RETURN_IF_ERROR(while_op_->input_edges(&edges)); in CreateEnterNodes()