Searched refs:input_edges (Results 1 – 17 of 17) 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()
104 inline InputEdges input_edges();484 Node::InputEdges Node::input_edges() { in input_edges() function
175 for (Edge const edge : node->input_edges()) { in UpdatePlacement()1275 stack.push(node->input_edges().begin()); in PrepareUses()1282 if (++stack.top() == edge.from()->input_edges().end()) stack.pop(); in PrepareUses()1286 if (node->InputCount() > 0) stack.push(node->input_edges().begin()); in PrepareUses()
454 for (Edge edge : this->input_edges()) { in Verify()
110 std::vector<const Edge*> input_edges; in ReorderInputEdges() local115 input_edges.push_back(e); in ReorderInputEdges()117 for (const Edge* e : input_edges) { in ReorderInputEdges()406 std::vector<const Edge*> input_edges; in MaybeRewriteIfNode() local411 input_edges.push_back(e); in MaybeRewriteIfNode()413 for (const Edge* e : input_edges) { in MaybeRewriteIfNode()
245 TF_RETURN_IF_ERROR(n->input_edges(&input_edges_vector)); in GetXlaClusterInfo()246 absl::Span<const Edge*> input_edges(input_edges_vector); in GetXlaClusterInfo() local248 absl::c_transform(input_edges.subspan(0, num_constant_inputs), in GetXlaClusterInfo()253 input_edges.subspan(num_constant_inputs, num_non_constant_inputs), in GetXlaClusterInfo()257 input_edges.subspan(num_constant_inputs + num_non_constant_inputs, in GetXlaClusterInfo()
263 TF_RETURN_IF_ERROR(launch->input_edges(&in_edges)); in BuildXlaLaunchOps()
281 Status Node::input_edges(std::vector<const Edge*>* input_edges) const { in input_edges() function in tensorflow::Node282 input_edges->clear(); in input_edges()283 input_edges->resize(num_inputs(), nullptr); in input_edges()290 if ((*input_edges)[edge->dst_input()] != nullptr) { in input_edges()294 (*input_edges)[edge->dst_input()] = edge; in input_edges()298 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()
215 TF_CHECK_OK(n->input_edges(&edges)); in InsertInputConversionNode()
214 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()
185 TF_RETURN_IF_ERROR(if_op_->input_edges(&edges)); in AddInputs()
176 TF_RETURN_IF_ERROR(case_op_->input_edges(&edges)); in AddInputs()
240 TF_RETURN_IF_ERROR(while_op_->input_edges(&edges)); in CreateEnterNodes()
1094 std::vector<const Edge*> input_edges; in IsTensorRTCandidate() local1095 TF_RETURN_IF_ERROR(node->input_edges(&input_edges)); in IsTensorRTCandidate()1096 for (const Edge* edge : input_edges) { in IsTensorRTCandidate()